Update app.py

app.py

@@ -41,8 +41,8 @@ def generate_text(prompt):
         inputs = tokenizer(f"<|STARTOFTEXT|> <|USER|> {prompt} <|BOT|>", return_tensors="pt").to(device)
         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 = generate_text.split("<|ENDOFTEXT|>")
-        print(generate_text)
+        result2 = generated_text.split("<|ENDOFTEXT|>")
+        print(generated_text)
     return result2[0]
 
 # Set up the Gradio interface