UmerSajid commited on
Commit
bb76140
·
verified ·
1 Parent(s): 3b79afe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -2,16 +2,18 @@ import streamlit as st
2
  import pandas as pd
3
  from transformers import MarianMTModel, MarianTokenizer
4
 
 
5
 
 
6
  MODEL_NAME = os.environ.get("model_name")
7
 
8
  if not MODEL_NAME:
9
  raise ValueError("MODEL_NAME is not set. Please add it in the Hugging Face Secrets.")
 
10
 
11
 
12
-
13
- model = MarianMTModel.from_pretrained(MODEL_NAME)
14
- tokenizer = MarianTokenizer.from_pretrained(MODEL_NAME)
15
 
16
  # Function to translate text from English to Urdu
17
  def translate_text(text):
 
2
  import pandas as pd
3
  from transformers import MarianMTModel, MarianTokenizer
4
 
5
+ model_name = 'Helsinki-NLP/opus-mt-en-ur'
6
 
7
+ """"
8
  MODEL_NAME = os.environ.get("model_name")
9
 
10
  if not MODEL_NAME:
11
  raise ValueError("MODEL_NAME is not set. Please add it in the Hugging Face Secrets.")
12
+ """
13
 
14
 
15
+ model = MarianMTModel.from_pretrained(model_name)
16
+ tokenizer = MarianTokenizer.from_pretrained(model_name)
 
17
 
18
  # Function to translate text from English to Urdu
19
  def translate_text(text):