hugging2021 commited on
Commit
6ca1a49
·
verified ·
1 Parent(s): 85b39f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -15,8 +15,10 @@ io2 = gr.load("huggingface/bert-base-uncased")
15
  def inference(inputtext, model):
16
  if model == "bert-base-cased":
17
  outlabel = io1(inputtext)
18
- else:
19
  outlabel = io2(inputtext)
 
 
20
  # Sicherstellen, dass die Ausgabe eine formatierte Zeichenkette ist
21
  return str(outlabel)
22
 
@@ -24,7 +26,7 @@ def inference(inputtext, model):
24
  gr.Interface(
25
  fn=inference,
26
  inputs=[
27
- gr.Textbox(label="Context", lines=10),
28
  gr.Dropdown(choices=["bert-base-cased", "bert-base-uncased"], value="bert-base-cased", label="model")
29
  ],
30
  outputs=gr.Label(label="Output"),
 
15
  def inference(inputtext, model):
16
  if model == "bert-base-cased":
17
  outlabel = io1(inputtext)
18
+ elif model == "bert-base-uncased":
19
  outlabel = io2(inputtext)
20
+ else:
21
+ outlabel = "Invalid model selected"
22
  # Sicherstellen, dass die Ausgabe eine formatierte Zeichenkette ist
23
  return str(outlabel)
24
 
 
26
  gr.Interface(
27
  fn=inference,
28
  inputs=[
29
+ gr.Textbox(label="Context", lines=10, placeholder="Enter text with [MASK] token"),
30
  gr.Dropdown(choices=["bert-base-cased", "bert-base-uncased"], value="bert-base-cased", label="model")
31
  ],
32
  outputs=gr.Label(label="Output"),