Spaces:

Someman
/

bloomz-nepali

Runtime error

Someman commited on Jan 8, 2024

Commit

06b4881

1 Parent(s): c14636b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def generate_output(prompt, input, kwargs):
     output = tokenizer.batch_decode(
         generate[:, inputs.input_ids.shape[1] :], skip_special_tokens=True
     )
-    return output[0].split("\n\n")[0].strip()
 def summarization(input: str):

     output = tokenizer.batch_decode(
         generate[:, inputs.input_ids.shape[1] :], skip_special_tokens=True
     )
+    return output[0].split("\n")[0].strip()
 def summarization(input: str):