Update app.py

app.py

@@ -2,7 +2,6 @@ import subprocess
 subprocess.run('pip install -r requirements.txt', shell = True)
 
 import gradio as gr
-import cv2
 import os
 from PIL import Image
 import numpy as np
@@ -88,8 +87,8 @@ def ragChain():
 
     #APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
     #os.environ["FIREWORKS_API_KEY"] = APIKEY
-
-    llm = HuggingFacePipeline.from_model_id(model_id=repo_id, task="text-generation")
+    model_id = "mosaicml/mpt-30b-chat"
+    llm = HuggingFacePipeline.from_model_id(model_id=repo_id, task="question-answering")
 
     prompt = ChatPromptTemplate.from_messages(
         [