xMcLovinx commited on
Commit
60095c1
·
verified ·
1 Parent(s): df9f949

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
3
  model = MBartForConditionalGeneration.from_pretrained("SnypzZz/Llama2-13b-Language-translate")
4
  tokenizer = MBart50TokenizerFast.from_pretrained("SnypzZz/Llama2-13b-Language-translate", src_lang="en_XX")
5
 
6
- iface = gr.Interface(fn=d, inputs=["textbox", gr.Dropdown(["de_DE", "es_XX"], label="Choose Output Language")], outputs="textbox")
7
 
8
  def d(input):
9
  model_inputs = tokenizer(input, return_tensors="pt")
@@ -15,4 +15,5 @@ def d(input):
15
  output_DE = output_DE.strip("[]' ")
16
  return output_DE
17
 
 
18
  iface.launch()
 
3
  model = MBartForConditionalGeneration.from_pretrained("SnypzZz/Llama2-13b-Language-translate")
4
  tokenizer = MBart50TokenizerFast.from_pretrained("SnypzZz/Llama2-13b-Language-translate", src_lang="en_XX")
5
 
6
+ dropdown = gr.Dropdown(["de_DE", "es_XX"], label="Choose Output Language")
7
 
8
  def d(input):
9
  model_inputs = tokenizer(input, return_tensors="pt")
 
15
  output_DE = output_DE.strip("[]' ")
16
  return output_DE
17
 
18
+ iface = gr.Interface(fn=d, inputs=["textbox", dropdown], outputs="textbox")
19
  iface.launch()