Fixing the output

app.py

@@ -41,6 +41,10 @@ def generate_response(instruction, input_text):
     # Decode the output into human-readable text
     generated_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
     
+    output_start = generated_text.find("### Output:")
+    if output_start != -1:
+        generated_text = generated_text[output_start + len("### Output:"):].strip()
+
     return generated_text
 
 # Gradio Interface with two inputs: Instruction and Input Text