Spaces:

nprasad24
/

Professor-Bean

Runtime error

nprasad24 commited on Jun 27, 2024

Commit

af4c500

verified ·

1 Parent(s): 6095473

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,7 +87,7 @@ def ragChain():
     #APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
     #os.environ["FIREWORKS_API_KEY"] = APIKEY
-    model_id = "mosaicml/mpt-30b-chat"
     llm = HuggingFacePipeline.from_model_id(model_id = model_id, task = "text-generation")
     prompt = ChatPromptTemplate.from_messages(

     #APIKEY = "o7T3gVx9Vt8GSJbLyPV1974vF8LXVp01CWqOkWQuHgoHm07H"
     #os.environ["FIREWORKS_API_KEY"] = APIKEY
+    model_id = "mosaicml/mpt-1b-redpajama-200b"
     llm = HuggingFacePipeline.from_model_id(model_id = model_id, task = "text-generation")
     prompt = ChatPromptTemplate.from_messages(