Spaces:

kishanmaharaj
/

cfilt-reference-based-hallucination-detection

Sleeping

App Files Files Community

kishanmaharaj commited on Jun 28

Commit

5e2bd12

·

verified ·

1 Parent(s): b7c6d1b

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ model_name = "MoritzLaurer/DeBERTa-v3-large-mnli-fever-anli-ling-wanli" #"Moritz
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
-device = torch.device("cuda" if torch.cuda.is_available() else "CPU")
 labels = ["entailment", "neutral", "contradiction"]

 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 labels = ["entailment", "neutral", "contradiction"]