MatteoScript commited on
Commit
1e929f5
·
1 Parent(s): 0a9fba8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
4
  app = FastAPI()
5
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
6
 
7
- class InputData(BaseModel):
8
  input: str
9
 
10
  def format_prompt(message, history):
 
4
  app = FastAPI()
5
  client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
6
 
7
+ class InputData():
8
  input: str
9
 
10
  def format_prompt(message, history):