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"),
|