mrgmd01 commited on
Commit
ca42861
·
verified ·
1 Parent(s): 77f6f6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,5 +1,9 @@
1
  import gradio as gr
 
2
  from transformers import pipeline
 
 
 
3
 
4
  english_model = pipeline("text-classification", model="mrgmd01/Finetuned_siebert-sentiment-roberta-large-english")
5
  urdu_model = pipeline("text-classification", model="mrgmd01/SA_Model_bert-base-multilingual-uncased")
 
1
  import gradio as gr
2
+ import os
3
  from transformers import pipeline
4
+ from huggingface_hub import login
5
+
6
+ login(token=os.environ.get("HF_TOKEN"))
7
 
8
  english_model = pipeline("text-classification", model="mrgmd01/Finetuned_siebert-sentiment-roberta-large-english")
9
  urdu_model = pipeline("text-classification", model="mrgmd01/SA_Model_bert-base-multilingual-uncased")