Update app.py

app.py

@@ -1,5 +1,27 @@
-import gradio as gr
+mport gradio as gr
 
-demo = gr.load("sinonimayzer/UzRoBERTa-v1", src="models")
+examples = [
+    ['The goal of life is <mask>.','']
+]
 
-demo.launch()
+uzroberta_v1 = gr.Interface.load("sinonimayzer/UzRoBERTa-v1")
+roberta_1_8_v2 = gr.Interface.load("sinonimayzer/roberta-1.8-v2")
+
+
+def inference(input, model):
+    if model == "UzRoBERTa-v1":
+        result = uzroberta_v1(input)
+    else:
+        result = roberta_1_8_v2(input)
+    return result
+
+gr.Interface(
+    inference, 
+    [
+        gr.inputs.Textbox(label="Context", lines=10),
+        gr.inputs.Dropdown(choices=["UzRoBERTa-v1","roberta-1.8-v2"], type="value", default="UzRoBERTa-v1", label="model")
+    ], 
+    [
+        gr.outputs.Label(label="Output")
+    ],
+    examples=examples).launch(enable_queue=True)