Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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":
|