Update app.py

app.py

@@ -6,22 +6,33 @@ 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)
+# 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")
+# 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])
+#     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)
+# if __name__ == "__main__":
+#     demo.queue().launch(share=True)
+
+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}
+
+demo = gr.Interface(fn=combine_models, inputs="text", outputs=["label", "label"])
+demo.launch()