testing

app.py

@@ -1,3 +1,20 @@
+import os
+import gradio as gr
+
+download="wget --load-cookies /tmp/cookies.txt \"https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1-hzy09qi-OEogyge7rQG79K7iV4xsNWa' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\1\\n/p')&id=1-hzy09qi-OEogyge7rQG79K7iV4xsNWa\" -O indic-en.zip && rm -rf /tmp/cookies.txt"
+os.system(download)
+os.system('unzip /home/user/app/indic-en.zip')
+
+from fairseq import checkpoint_utils, distributed_utils, options, tasks, utils
+from inference.engine import Model
+indic2en_model = Model(expdir='/home/user/app/indic-en')
+
+INDIC = {"Assamese": "as", "Bengali": "bn", "Gujarati": "gu", "Hindi": "hi","Kannada": "kn","Malayalam": "ml", "Marathi": "mr", "Odia": "or","Punjabi": "pa","Tamil": "ta", "Telugu" : "te"}
+
+
+def translate(text, lang):
+  return indic2en_model.translate_paragraph(text, INDIC[lang], 'en')
+
 from transformers import pipeline
 import gradio as gr
 roberta_pipe = pipeline(
@@ -7,22 +24,9 @@ roberta_pipe = pipeline(
     return_all_scores = True
 )
 
-import requests
-
-TRANS_API = "https://hussain-shk-indictrans-indic2english.hf.space/run/predict"
-
-def trans(text: str, source:str, target:str):
-    response = requests.post(TRANS_API, json={
-    "data": [
-      text,
-      source,
-    ]
-  }, timeout=None).json()
-    return response["data"][0]
-
 def analyse_sentiment(text, source):
    if source != "English":
-        text = trans(text, source, "English")
+        text = translate(text, source, "English")
    response = roberta_pipe(text)
    d = {}
    for i in response[0]:
@@ -38,7 +42,7 @@ examples = [["this book was a great book that i have read many times", "English"
               ["एक महान अमेरिकी लेखक का एक आकर्षक संग्रह" , "Hindi"],
               ["हा आतापर्यंतचा सर्वात वाईट चित्रपट आहे यात शंका नाही", "Marathi"],
               ["இந்த தயாரிப்பு ஆச்சரியமாக இருக்கிறது", "Tamil"],
-              ["તમારા માટે નહીં જો તમે વિના અવરોધે વીડિયો શોધી રહ્યા છો", "Gujrati"],]
+              ["તમારા માટે નહીં જો તમે વિના અવરોધે વીડિયો શોધી રહ્યા છો", "Gujarati"],]
 
 demo = gr.Interface(
     enable_queue=True,
@@ -49,4 +53,4 @@ demo = gr.Interface(
     title='IndiSent: Multilingual Sentiment Analysis',
     examples=examples)
 
-demo.launch()
+demo.launch(share = True, debug=True, inline=True)