Hugging Face's logo

Spaces:
ADKU
/
ResearchGPT_space
Sleeping

App Files Community
ADKU commited on
Commit
7f97f1b
·
verified ·
1 Parent(s): 8220e51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import torch
4
 
5
  app = FastAPI()
6
 
7
- model = DistilBertForSequenceClassification.from_pretrained("your-username/distilbert-recommendation")
8
- tokenizer = DistilBertTokenizerFast.from_pretrained("your-username/distilbert-recommendation")
9
 
10
  @app.post("/predict/")
11
  async def predict(text: str):
 
4
 
5
  app = FastAPI()
6
 
7
+ model = DistilBertForSequenceClassification.from_pretrained("ADKU/ResearchGPT_model")
8
+ tokenizer = DistilBertTokenizerFast.from_pretrained("ADKU/ResearchGPT_model")
9
 
10
  @app.post("/predict/")
11
  async def predict(text: str):