tangminhanh commited on
Commit
7d825dd
·
verified ·
1 Parent(s): 0f499af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
1
  from fastapi import FastAPI, Query
2
  from pydantic import BaseModel
3
  from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
4
 
5
  app = FastAPI()
6
- token = os.getenv("HF_TOKEN")
7
 
8
  # Load the model and tokenizer
9
  model = AutoModelForSequenceClassification.from_pretrained(
 
1
  from fastapi import FastAPI, Query
2
  from pydantic import BaseModel
3
  from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
4
+ import os
5
 
6
  app = FastAPI()
7
+ token = os.getenv("token")
8
 
9
  # Load the model and tokenizer
10
  model = AutoModelForSequenceClassification.from_pretrained(