updating again

app.py

@@ -8,6 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Function to generate a response from GPT-2
 def generate_response(prompt):
+    # Only generate a response based on the given prompt
     inputs = tokenizer(prompt, return_tensors="pt")
     output = model.generate(inputs['input_ids'], max_length=150)
     response = tokenizer.decode(output[0], skip_special_tokens=True)
@@ -23,7 +24,7 @@ st.write("This chatbot is your mentor to help you with learning Data Structures.
 user_input = st.text_input("You:", "")
 
 if user_input:
-    # Adding context to the prompt, making GPT-2 respond like a mentor
+    # Adding context to the prompt, but only once for the first input
     prompt = f"You are a mentor teaching data structures. Answer the following question: {user_input}"
     response = generate_response(prompt)
     st.text_area("Mentor's Response:", value=response, height=200, disabled=True)