Update app.py

app.py

@@ -1,18 +1,18 @@
+import os
 import gradio as gr
 from transformers import pipeline
 
-# Token do Hugging Face para modelos privados
-hf_token = "hf_SEU_TOKEN_AQUI"
+# Pegar token do secret
+hf_token = os.environ["HF_TOKEN"]
 
-# Carregar modelo privado
 generator = pipeline(
-    "text-generation", 
-    model="melyssa08/model_collapse_generation_0", 
+    "text-generation",
+    model="melyssa08/model_collapse_generation_0",
     use_auth_token=hf_token
 )
 
 def gerar_texto(texto):
     return generator(texto, max_length=50)[0]["generated_text"]
 
-# Interface Gradio
 gr.Interface(gerar_texto, "text", "text").launch()
+