nprasad24 commited on
Commit
6095473
·
verified ·
1 Parent(s): 66899d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def ragChain():
88
  #APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
89
  #os.environ["FIREWORKS_API_KEY"] = APIKEY
90
  model_id = "mosaicml/mpt-30b-chat"
91
- llm = HuggingFacePipeline.from_model_id(model_id = model_id, task = "question-answering")
92
 
93
  prompt = ChatPromptTemplate.from_messages(
94
  [
 
88
  #APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
89
  #os.environ["FIREWORKS_API_KEY"] = APIKEY
90
  model_id = "mosaicml/mpt-30b-chat"
91
+ llm = HuggingFacePipeline.from_model_id(model_id = model_id, task = "text-generation")
92
 
93
  prompt = ChatPromptTemplate.from_messages(
94
  [