more fixes!

app.py

@@ -1,12 +1,13 @@
 import numpy as np
 import gradio as gr
-from transformers import TFAutoModelForSequenceClassification, AutoTokenizer
+import torch
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
 checkpoint="MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli"
 
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model=TFAutoModelForSequenceClassification.from_pretrained(checkpoint,from_pt=True)
+model=AutoModelForSequenceClassification.from_pretrained(checkpoint)
 
 title = "The Seagull story"
 description = """
@@ -64,8 +65,10 @@ def generate_tone(index,question):
     """.replace("\n","")
 
     ]
-    inputs=tokenizer(passages[index],question, return_tensors="tf")
-    return (model(**inputs).logits)
+    input = tokenizer(passages[index], question, truncation=True, return_tensors="pt")
+    output = model(input["input_ids"].to("cpu"))  # device = "cuda:0" or "cpu"
+    prediction = torch.softmax(output["logits"][0], -1).tolist()
+    return prediction
 
 passages=["General","Pier","Boat","Island"]
 

requirements.txt

@@ -1,3 +1,3 @@
 transformers
 sentencepiece
-tensorflow
+torch