Hugging Face's logo

Spaces:
manuu01
/
SeagullStory
Sleeping

App Files Community
manuu01 commited on
Commit
4100561
•
1 Parent(s): 4d26442

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,11 +3,11 @@ import gradio as gr
3
  import torch
4
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
 
6
- checkpoint="MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli"
7
 
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
10
- model=AutoModelForSequenceClassification.from_pretrained(checkpoint)
11
  answers=['Yes',"Doesn't matter","No"]
12
  f=open("Questions.txt","a")
13
 
 
3
  import torch
4
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
 
6
+ checkpoint="manuu01/DeBERTa-SeagullStory"
7
 
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
10
+ model=AutoModelForSequenceClassification.from_pretrained(checkpoint,torch_dtype=torch.float16)
11
  answers=['Yes',"Doesn't matter","No"]
12
  f=open("Questions.txt","a")
13