Update app.py

app.py

@@ -6,14 +6,13 @@ model = AutoModelForSeq2SeqLM.from_pretrained(".", local_files_only=True)
 pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
 
 def predict(task, prompt, context="", auto_cot=False):
-    input_str = f"[TASK: {task.upper()}] {prompt}"
+    full_prompt = f"[TASK: {task.upper()}] {prompt}"
     if context:
-        input_str += f" Context: {context}"
+        full_prompt += f" Context: {context}"
     if auto_cot:
-        input_str += "\nLet's think step by step."
-
-    result = pipe(input_str, max_new_tokens=128)[0]["generated_text"]
-    return result
+        full_prompt += "\nLet's think step by step."
+    output = pipe(full_prompt, max_new_tokens=128)[0]["generated_text"]
+    return output
 
 demo = gr.Interface(
     fn=predict,