BoxzDev commited on
Commit
79b886d
·
verified ·
1 Parent(s): 708841b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -7,8 +7,8 @@ import uvicorn # Import uvicorn for running the FastAPI application
7
  app = FastAPI() # Create a FastAPI instance
8
 
9
  # Define the primary and fallback models
10
- primary = "HuggingFaceH4/zephyr-7b-beta"
11
- fallbacks = ["mistralai/Mistral-7B-Instruct-v0.2", "HuggingFaceH4/zephyr-7b-beta"]
12
 
13
  # Define the data model for the request body
14
  class Item(BaseModel):
 
7
  app = FastAPI() # Create a FastAPI instance
8
 
9
  # Define the primary and fallback models
10
+ primary = "mistralai/Mixtral-8x7B-Instruct-v0.1"
11
+ fallbacks = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1"]
12
 
13
  # Define the data model for the request body
14
  class Item(BaseModel):