Spaces:
Sleeping
Sleeping
zhenyundeng
commited on
Commit
•
075c300
1
Parent(s):
0d26419
update
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from fastapi import FastAPI
|
|
| 6 |
from pydantic import BaseModel
|
| 7 |
# from averitec.models.AveritecModule import Wikipediaretriever, Googleretriever, veracity_prediction, justification_generation
|
| 8 |
import uvicorn
|
| 9 |
-
import spaces
|
| 10 |
|
| 11 |
app = FastAPI()
|
| 12 |
|
|
@@ -75,6 +75,7 @@ LABEL = [
|
|
| 75 |
]
|
| 76 |
|
| 77 |
# Veracity
|
|
|
|
| 78 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 79 |
veracity_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
|
| 80 |
bert_model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=4, problem_type="single_label_classification")
|
|
|
|
| 6 |
from pydantic import BaseModel
|
| 7 |
# from averitec.models.AveritecModule import Wikipediaretriever, Googleretriever, veracity_prediction, justification_generation
|
| 8 |
import uvicorn
|
| 9 |
+
# import spaces
|
| 10 |
|
| 11 |
app = FastAPI()
|
| 12 |
|
|
|
|
| 75 |
]
|
| 76 |
|
| 77 |
# Veracity
|
| 78 |
+
|
| 79 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 80 |
veracity_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
|
| 81 |
bert_model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=4, problem_type="single_label_classification")
|