Update app.py

app.py

@@ -13,7 +13,7 @@ from transformers import pipeline
 stt = pipeline("automatic-speech-recognition", model="openai/whisper-small")
 
 # Better Tutor Model (FLAN-T5)
-llm = pipeline("text2text-generation", model="google/flan-t5-small")
+llm = pipeline("text-generation", model="distilgpt2")
 
 
 # -------------------------------
@@ -37,26 +37,22 @@ def speech_to_text(audio):
 
 
 def generate_response(text):
-    """
-    Generates tutor-style response
-    """
     if not text or text == "No audio provided.":
         return "Please provide valid input."
 
     prompt = f"""
-    You are an expert AI tutor.
-
-    Explain:
-    - in simple words
-    - step by step
-    - with examples if possible
+    You are an AI tutor.
+    Explain clearly and simply.
 
     Question: {text}
     Answer:
     """
 
-    output = llm(prompt, max_length=150)
-    return output[0]["generated_text"]
+    output = llm(prompt, max_length=150, num_return_sequences=1)
+    response = output[0]["generated_text"]
+
+    # Clean output
+    return response.split("Answer:")[-1].strip()
 
 
 # -------------------------------