Upload app.py

app.py

@@ -91,68 +91,67 @@ if __name__ == "__main__":
     demo = gr.Blocks()
     with demo:
         with gr.Row():
-            if triplet_extractor:
-                with gr.Column():
-                    gr.Markdown(
-                        "# <p align='center'>Hospital Review Aspect Sentiment Generation</p>"
-                    )
-                    with gr.Row():
-                        with gr.Column():
-                            asde_input_sentence = gr.Textbox(
-                                placeholder="Leave this box blank and choose a dataset will give you a random example...",
-                                label="Example:",
-                            )
-                            gr.Markdown(
-                                "You can find code and dataset at [MTech Thesis Project](https://github.com/Amir22010)"
-                            )
-                            asde_dataset_ids = gr.Radio(
-                                choices=[
-                                    "HospitalReviews"
-                                ],
-                                value="HospitalReviews",
-                                label="Datasets",
-                            )
-                            
-                            asde_model_ids = gr.Radio(
-                                choices=[
-                                    "PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
-                                   # "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
-                                   # "PyABSA_Hospital_KeyBert_allenai/tk-instruct-base-def-pos_FinedTuned_Model"
-                                ],
-                                value="PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model",
-                                label="Fine-tuned Models on Hospital Review custom data",
-                            )
-
-                            asde_inference_button = gr.Button("Let's go!")
-
-                            asde_output_text = gr.TextArea(label="Example:")
-
-                            asde_model_output_generated_sentence = gr.Textbox(
-                                placeholder="Text Generated...",
-                                label="Model Prediction Text Generated:",
-                            )
-                            asde_output_pred_df = gr.DataFrame(
-                                label="Predicted Aspect & Sentiment:"
-                            )
-                            asde_output_true_df = gr.DataFrame(
-                                label="Original Aspect & Sentiment:"
-                            )
-
-                            asde_inference_button.click(
-                                fn=run_demo,
-                                inputs=[
-                                    asde_input_sentence,
-                                    asde_dataset_ids,
-                                    asde_model_ids,
-                                    gr.Text("ASDE", visible=False),
-                                ],
-                                outputs=[
-                                    asde_output_pred_df,
-                                    asde_output_true_df,
-                                    asde_output_text,
-                                    asde_model_output_generated_sentence
-                                ],
-                            )
+            with gr.Column():
+                gr.Markdown(
+                    "# <p align='center'>Hospital Review Aspect Sentiment Generation</p>"
+                )
+                with gr.Row():
+                    with gr.Column():
+                        asde_input_sentence = gr.Textbox(
+                            placeholder="Leave this box blank and choose a dataset will give you a random example...",
+                            label="Example:",
+                        )
+                        gr.Markdown(
+                            "You can find code and dataset at [MTech Thesis Project](https://github.com/Amir22010)"
+                        )
+                        asde_dataset_ids = gr.Radio(
+                            choices=[
+                                "HospitalReviews"
+                            ],
+                            value="HospitalReviews",
+                            label="Datasets",
+                        )
+                        
+                        asde_model_ids = gr.Radio(
+                            choices=[
+                                "PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
+                                # "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model", 
+                                # "PyABSA_Hospital_KeyBert_allenai/tk-instruct-base-def-pos_FinedTuned_Model"
+                            ],
+                            value="PyABSA_Hospital_English_allenai/tk-instruct-base-def-pos_FinedTuned_Model",
+                            label="Fine-tuned Models on Hospital Review custom data",
+                        )
+
+                        asde_inference_button = gr.Button("Let's go!")
+
+                        asde_output_text = gr.TextArea(label="Example:")
+
+                        asde_model_output_generated_sentence = gr.Textbox(
+                            placeholder="Text Generated...",
+                            label="Model Prediction Text Generated:",
+                        )
+                        asde_output_pred_df = gr.DataFrame(
+                            label="Predicted Aspect & Sentiment:"
+                        )
+                        asde_output_true_df = gr.DataFrame(
+                            label="Original Aspect & Sentiment:"
+                        )
+
+                        asde_inference_button.click(
+                            fn=run_demo,
+                            inputs=[
+                                asde_input_sentence,
+                                asde_dataset_ids,
+                                asde_model_ids,
+                                gr.Text("ASDE", visible=False),
+                            ],
+                            outputs=[
+                                asde_output_pred_df,
+                                asde_output_true_df,
+                                asde_output_text,
+                                asde_model_output_generated_sentence
+                            ],
+                        )
         gr.Markdown(
             """### GitHub Repo: [MTech Thesis Project](https://github.com/Amir22010)
             ### Author: [Amir Khan](https://github.com/Amir22010)