mrgmd01 commited on
Commit
781ea3c
·
verified ·
1 Parent(s): ca42861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,12 +1,11 @@
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")
10
 
11
  def predict(text, language):
12
  if language == "urdu" or language == "roman_urdu":
 
1
  import gradio as gr
2
  import os
3
  from transformers import pipeline
 
4
 
5
+ token = os.environ.get("HF_TOKEN")
6
 
7
+ english_model = pipeline("text-classification", model="mrgmd01/Finetuned_siebert-sentiment-roberta-large-english", token=token)
8
+ urdu_model = pipeline("text-classification", model="mrgmd01/SA_Model_bert-base-multilingual-uncased", token=token)
9
 
10
  def predict(text, language):
11
  if language == "urdu" or language == "roman_urdu":