Update app.py

app.py

@@ -1,18 +1,37 @@
 import os
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-# Pegar token do secret
+# ✅ Pegar token do Hugging Face do Secret
 hf_token = os.environ["HF_TOKEN"]
 
-generator = pipeline(
-    "text-generation",
-    model="melyssa08/model_collapse_generation_0",
-    use_auth_token=hf_token
-)
+# Nome do modelo privado
+model_name = "melyssa08/model_collapse_generation_0"
 
+# ✅ Carregar tokenizer e modelo privado
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token)
+
+# ✅ Criar pipeline de geração de texto
+generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+# Função para gerar texto
 def gerar_texto(texto):
-    return generator(texto, max_length=50)[0]["generated_text"]
+    if not texto:
+        return "Por favor, forneça algum texto."
+    result = generator(texto, max_length=50, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+# ✅ Interface Gradio
+# Pode usar Blocks para permitir POST JSON
+with gr.Blocks() as demo:
+    gr.Markdown("### Chatbot de Texto com seu modelo privado")
+    input_text = gr.Textbox(label="Digite seu texto")
+    output_text = gr.Textbox(label="Texto gerado")
+    btn = gr.Button("Gerar")
+    btn.click(gerar_texto, inputs=input_text, outputs=output_text)
+
+# ✅ Rodar Gradio com share=True para gerar URL pública
+demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
 
-gr.Interface(gerar_texto, "text", "text").launch()