Update app.py

app.py

@@ -3,14 +3,20 @@ from transformers import pipeline
 import os
 
 # Modelos leves recomendados
+
+
 MODEL_OPTIONS = {
     "Qwen2.5:3B": "Qwen/Qwen2.5-3B-Instruct",
     "GPT-2 Médio": "gpt2-medium",
-    "BLOOM (560M)": "bigscience/bloom-560m"
+    "BLOOM (560M)": "bigscience/bloom-560m",
+    "DistilGPT2": "distilgpt2",
+    "TinyLLaMA": "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 }
 
+
+
 # Escolha o modelo Qwen2.5
-MODEL_NAME = MODEL_OPTIONS["Qwen2.5:3B"]
+MODEL_NAME = MODEL_OPTIONS["TinyLLaMA"]
 
 def load_model():
     global MODEL_NAME  # Declarar global no início da função