Update app.py

app.py

@@ -8,7 +8,8 @@ model = AutoModelForCausalLM.from_pretrained("gpt-chatbot")
 
 
 def generate_response(input_text):
-    input_ids = tokenizer.encode(input_text + " >> User: ", return_tensors='pt')
+    input_text = "User: " + input_text
+    input_ids = tokenizer.encode(input_text, return_tensors='pt')
     generated_output = model.generate(input_ids, max_length=100, num_return_sequences=1)
     response = tokenizer.decode(generated_output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
     return response