Update app.py
Browse files
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 |
-
|
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"),
|