Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,8 @@ app = FastAPI()
|
|
6 |
|
7 |
hf_token = os.getenv('HUGGINGFACE_HUB_TOKEN')
|
8 |
|
9 |
-
model = AutoModelForCausalLM.from_pretrained("
|
10 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
11 |
|
12 |
@app.post("/inference")
|
13 |
async def inference(prompt: str):
|
|
|
6 |
|
7 |
hf_token = os.getenv('HUGGINGFACE_HUB_TOKEN')
|
8 |
|
9 |
+
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.3", use_auth_token=hf_token)
|
10 |
+
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.3")
|
11 |
|
12 |
@app.post("/inference")
|
13 |
async def inference(prompt: str):
|