Spaces:

Iker
/

ClickbaitFighter

Running on Zero

Iker commited on Jul 24, 2024

Commit

518ecc9

1 Parent(s): 83aa4d1

Fix

Files changed (1) hide show

app_zero.py CHANGED Viewed

@@ -149,7 +149,7 @@ def run_model(mode, title, text):
         logits_processor=stop_criteria,
     )
-    yield title, streamer, text
     temp = tokenizer.batch_decode(
         model_output[:, model_inputs["input_ids"].shape[-1] :],

         logits_processor=stop_criteria,
     )
+    yield streamer
     temp = tokenizer.batch_decode(
         model_output[:, model_inputs["input_ids"].shape[-1] :],