Update app.py

app.py

@@ -42,9 +42,9 @@ def generate_text(prompt):
         outputs = model.generate(inputs['input_ids'], max_length=100, num_return_sequences=1, do_sample=False)
         generated_text = tokenizer.decode(outputs[0], skip_special_tokens=False)
         result2 = generated_text.split("<|ENDOFTEXT|>")
-        finalRes = result2.split("<|BOT|>")
+        finalRes = result2[0].split("<|BOT|>")
         print(generated_text)
-    return finalRes[1]
+    return finalRes[-1]
 
 # Set up the Gradio interface
 iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="Text Generator using Fine-Tuned Model")