Update app.py

app.py

@@ -1,7 +1,7 @@
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import BartForSequenceClassification, BartTokenizer
 import gradio as grad
-tokenizer = AutoTokenizer.from_pretrained("oigele/Fb_improved_zeroshot")
-model = AutoModelForSequenceClassification.from_pretrained("oigele/Fb_improved_zeroshot")
+bart_tkn = BartTokenizer.from_pretrained('oigele/Fb_improved_zeroshot')
+mdl = BartForSequenceClassification.from_pretrained('oigele/Fb_improved_zeroshot')
 
 def classify(text,label): 
     tkn_ids = bart_tkn.encode(text, label, return_tensors='pt')