Update app.py

app.py

@@ -6,7 +6,7 @@ generator = pipeline("text-generation", model="foryui/Llama-3.2-1B-Instruct-GRPO
 
 def my_inference_function(text):
     prompt = f"Summary the context below\n\n{text}"
-    output = generator([{"role": "user", "content": question}], max_new_tokens=1024, return_full_text=False)[0]
+    output = generator([{"role": "user", "content": prompt}], max_new_tokens=1024, return_full_text=False)[0]
     output = output["generated_text"]
     
     return output[len(prompt):]