Update app.py

app.py

@@ -6,33 +6,28 @@ examples = [
     'Shu munosabat bilan O‘zbekiston Prezidenti global inqiroz sharoitida savdo-iqtisodiy hamkorlikni <mask> va hududlararo aloqalarni rivojlantirishning muhim masalalariga to‘xtalib o‘tdi.',
 ]
 
-# def fn(text):
-#     uzroberta_v1 = pipeline("fill-mask", model="sinonimayzer/UzRoBERTa-v1")
-#     roberta_1_8_v2 = pipeline("fill-mask", model="sinonimayzer/roberta-1.8-v2")
-#     return uzroberta_v1(text), roberta_1_8_v2(text)
-
-# with gr.Blocks() as demo:
-#     with gr.Row():
-#         with gr.Column():
-#             inp = gr.Textbox(label="Input")
-#         with gr.Column():
-#             output1 = gr.Label(label="UzRoBERTa-v1")
-#             output2 = gr.Label(label="roberta-1.8-v2")
+def df(arr):
+    d = {}
+    for val in arr:
+        d[val['token_str']] = val['score']
+    return d
     
-#     btn = gr.Button("Generate")
-#     btn.click(fn, inputs=[inp], outputs=[output1, output2])
-#     gr.Examples(examples, inputs=[inp])
-
-# if __name__ == "__main__":
-#     demo.queue().launch(share=True)
+def fn(text):
+    uzroberta_v1 = pipeline("fill-mask", model="sinonimayzer/UzRoBERTa-v1")
+    roberta_1_8_v2 = pipeline("fill-mask", model="sinonimayzer/roberta-1.8-v2")
+    return df(uzroberta_v1(text)), df(roberta_1_8_v2(text))
 
-m1 = gr.load("sinonimayzer/UzRoBERTa-v1", src="models")
-m2 = gr.load("sinonimayzer/roberta-1.8-v2", src="models")
-
-def combine_models(input_text):
-    output1 = m1.predict(input_text)
-    output2 = m2.predict(input_text)
-    return {"Model 1 Output": output1, "Model 2 Output": output2}
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            inp = gr.Textbox(label="Input")
+        with gr.Column():
+            output1 = gr.Label(label="UzRoBERTa-v1")
+            output2 = gr.Label(label="roberta-1.8-v2")
+    
+    btn = gr.Button("Generate")
+    btn.click(fn, inputs=[inp], outputs=[output1, output2])
+    gr.Examples(examples, inputs=[inp])
 
-demo = gr.Interface(fn=combine_models, inputs="text", outputs=["label", "label"])
-demo.launch()
+if __name__ == "__main__":
+    demo.queue().launch(share=True)