Update app.py

app.py

@@ -14,11 +14,14 @@ llm = LlamaCpp(
     )
 
 # Generator function
-def gen_quiz(input, num):
+def gen_quiz(text_input,num):
     prompt = ChatPromptTemplate.from_template(
-    "Generate {num} questions and their correct answers based on the following text:\n\n{text}\n\n")
+        "Generate {num} questions and their correct answers based on the following text:\n\n{text}\n\n"
+    )
+    # Prepare the inputs for the chain
+    prompt_input = {"text": text_input, "num": num}
     chain = LLMChain(llm=llm, prompt=prompt)    
-    quiz = chain.invoke(input)
+    quiz = chain(prompt_input)
     return quiz['text']
 
 # Example