Fix

app_zero.py

@@ -149,7 +149,7 @@ def run_model(mode, title, text):
         logits_processor=stop_criteria,
     )
 
-    yield streamer
+    # yield streamer # Does not work properly on Zero environment
 
     temp = tokenizer.batch_decode(
         model_output[:, model_inputs["input_ids"].shape[-1] :],
@@ -223,8 +223,8 @@ def generate_text(
         progress(0.5, desc="🤖 Leyendo noticia")
 
         try:
-            for temp in run_model(mode, title, text):
-                yield title, temp, text
+            temp = run_model(mode, title, text):
+            
 
         except Exception as e:
             print(e)