Update app.py

app.py

@@ -13,12 +13,18 @@ def generate_prompts(user_input):
     config = {'max_new_tokens': 2048, 'temperature': 0.7, 'context_length': 4096}
 
     llm = HuggingFaceHub(
-    repo_id="google/flan-t5-xxl", model_kwargs={"temperature": 0.5, "max_length": 64})    
+    repo_id="google/flan-t5-xxl", model_kwargs={"temperature": 0.5, "max_length": 64}) 
+
+    model_name = "deepset/roberta-base-squad2"
+    model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+    llm = pipeline("question-answering", model=model, tokenizer=tokenizer)
     
     hub_chain = LLMChain(prompt = prompt_template, llm = llm)
     input_data = {"Question": user_input}
 
-    generated_prompts = hub_chain.run(input_data)  # Modify this part based on how you run the model
+    generated_prompts = hub_chain.run(input_data)  
     questions_list = generated_prompts.split('\n')