Update app.py

app.py

@@ -20,7 +20,11 @@ def analyze_argument(argument):
     input_ids = tokenizer(prompt, return_tensors="pt")
 
     outputs = model.generate(**input_ids, max_length=1000, num_return_sequences=1)
-    return tokenizer.decode(outputs[0])
+    output = tokenizer.decode(outputs[0])
+    # remove prompt from output
+    output = output.split("### Response:")[1].strip().split("<|endoftext|>")[0]
+    return output
 
 
-gr.Interface(analyze_argument, inputs="text", outputs="text").launch()
+gr.Interface(analyze_argument, inputs="text", outputs="text",
+    examples=[["If it's wednesday it's cold, and it's cold, therefore it's wednesday."]]).launch()