Robadeldesouza commited on
Commit
20c46bc
·
verified ·
1 Parent(s): 175ac2b

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -56
app.py DELETED
@@ -1,56 +0,0 @@
1
- import os
2
- import gradio as gr
3
- from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
4
- import torch
5
-
6
- # Modelo leve para programação (Salesforce CodeGen)
7
- MODEL_NAME = "Salesforce/codegen-350M-mono"
8
- # Troque para "Salesforce/codegen-2B-mono" caso queira mais potência
9
-
10
- # Usa token se tiver (configure em .env ou nos Secrets do Space)
11
- HF_TOKEN = os.getenv("HF_TOKEN") # deixa em branco se não for usar
12
-
13
- # Carregar tokenizer e modelo (autenticação automática se HF_TOKEN presente)
14
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
15
- model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
16
-
17
- # Forçar uso de CPU no Spaces gratuito (device -1). Se houver GPU, mude para device=0
18
- device = 0 if torch.cuda.is_available() else -1
19
-
20
- generator = pipeline(
21
- "text-generation",
22
- model=model,
23
- tokenizer=tokenizer,
24
- device=device
25
- )
26
-
27
- def gerar_codigo(prompt):
28
- # Prompt simples — você pode customizar instruções do sistema aqui
29
- inputs = prompt.strip()
30
- try:
31
- out = generator(
32
- inputs,
33
- max_new_tokens=256,
34
- do_sample=True,
35
- temperature=0.2,
36
- top_p=0.9,
37
- num_return_sequences=1,
38
- pad_token_id=tokenizer.eos_token_id
39
- )
40
- text = out[0]["generated_text"]
41
- return text.strip()
42
- except Exception as e:
43
- return f"Erro durante geração: {e}"
44
-
45
- with gr.Blocks() as demo:
46
- gr.Markdown("""
47
- # 🤖 Jarvis - Assistente de Programação (CodeGen)
48
- **Instruções:** escreva sua solicitação de código no campo abaixo. Use linguagem natural.
49
- """)
50
- inp = gr.Textbox(lines=6, placeholder="Ex: Escreva uma função Python que calcula fatorial")
51
- out = gr.Textbox(lines=18, label="Saída do modelo")
52
- btn = gr.Button("Gerar")
53
- btn.click(fn=gerar_codigo, inputs=inp, outputs=out)
54
-
55
- if __name__ == '__main__':
56
- demo.launch()