AFischer1985 commited on
Commit
dd08ab3
·
1 Parent(s): 5ac8eb7

Add default to Dropdown

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  from transformers import pipeline
3
  title= "German Flan-T5"
4
  examples = [
5
- ["Erzähl mit eine Geschichte!",200,2,3,10,"Deutsch"]
 
6
  ]
7
 
8
  tDeEn = pipeline(model="Helsinki-NLP/opus-mt-de-en")
@@ -21,11 +22,11 @@ task = gr.Interface(
21
  fn=solve,
22
  inputs=[
23
  gr.Textbox(lines=5,max_lines=6,label="Frage"),
24
- gr.Slider(minimum=1.0,maximum=200.0,value=200.0,step=1,interactive=True,label="max_length"),
25
  gr.Slider(minimum=1.0,maximum=20.0,value=1.0,step=1,interactive=True,label="length_penalty"),
26
- gr.Slider(minimum=0.0,maximum=20.0,value=3.0,step=1,interactive=True,label="no_repeat_ngram_size"),
27
  gr.Slider(minimum=1.0,maximum=20.0,value=1.0,step=1,interactive=True,label="num_beams"),
28
- gr.Dropdown(["Deutsch", "Englisch"]),
29
  ],
30
  outputs="text",
31
  title=title,
 
2
  from transformers import pipeline
3
  title= "German Flan-T5"
4
  examples = [
5
+ ["Erzähl mit eine Geschichte!",50,2,3,1,"Deutsch"],
6
+ ["Please write a step by step recipe to make bolognese pasta!",50,2,3,2,"Englisch"]
7
  ]
8
 
9
  tDeEn = pipeline(model="Helsinki-NLP/opus-mt-de-en")
 
22
  fn=solve,
23
  inputs=[
24
  gr.Textbox(lines=5,max_lines=6,label="Frage"),
25
+ gr.Slider(minimum=1.0,maximum=200.0,value=50.0,step=1,interactive=True,label="max_length"),
26
  gr.Slider(minimum=1.0,maximum=20.0,value=1.0,step=1,interactive=True,label="length_penalty"),
27
+ gr.Slider(minimum=0.0,maximum=5.0,value=3.0,step=1,interactive=True,label="no_repeat_ngram_size"),
28
  gr.Slider(minimum=1.0,maximum=20.0,value=1.0,step=1,interactive=True,label="num_beams"),
29
+ gr.Dropdown(["Deutsch", "Englisch"],value="Deutsch"),
30
  ],
31
  outputs="text",
32
  title=title,