Update app.py
Browse files
app.py
CHANGED
|
@@ -13,13 +13,15 @@ io1 = gr.load("huggingface/bert-base-cased")
|
|
| 13 |
io2 = gr.load("huggingface/bert-base-uncased")
|
| 14 |
|
| 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 |
|
| 25 |
# Aktualisierte Gradio-Syntax
|
|
@@ -34,4 +36,4 @@ gr.Interface(
|
|
| 34 |
article=article,
|
| 35 |
title=title,
|
| 36 |
description=description
|
| 37 |
-
).launch()
|
|
|
|
| 13 |
io2 = gr.load("huggingface/bert-base-uncased")
|
| 14 |
|
| 15 |
def inference(inputtext, model):
|
| 16 |
+
if "[MASK]" not in inputtext:
|
| 17 |
+
return "Error: The input text must contain the [MASK] token."
|
| 18 |
+
|
| 19 |
if model == "bert-base-cased":
|
| 20 |
outlabel = io1(inputtext)
|
| 21 |
elif model == "bert-base-uncased":
|
| 22 |
outlabel = io2(inputtext)
|
| 23 |
else:
|
| 24 |
outlabel = "Invalid model selected"
|
|
|
|
| 25 |
return str(outlabel)
|
| 26 |
|
| 27 |
# Aktualisierte Gradio-Syntax
|
|
|
|
| 36 |
article=article,
|
| 37 |
title=title,
|
| 38 |
description=description
|
| 39 |
+
).launch(share=True)
|