solving problemas

app.py

@@ -7,7 +7,7 @@ inference = Inference()
 
 
 with gr.Blocks() as block:
-    options = gr.Dropdown(choices=["Blip Saffal", "Blip Control Net"], label="Models", info="Select the model to use..", )
+    options = gr.Dropdown(choices=["Blip Saffal", "Blip CN"], label="Models", info="Select the model to use..", )
     # need to improve this one...
 
     txt = gr.Textbox(label="Insert a question..", lines=2)

inference.py

@@ -6,20 +6,17 @@ class Inference:
         self.blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
         self.blip_model_saffal = BlipForQuestionAnswering.from_pretrained("wiusdy/blip_pretrained_saffal_fashion_finetuning")
         self.blip_model_control_net = BlipForQuestionAnswering.from_pretrained("wiusdy/blip_pretrained_control_net_fashion_finetuning")
-
         logging.set_verbosity_info()
         self.logger = logging.get_logger("transformers")
 
-    def inference(self, options, image, text):
-        if options == "Blip Saffal":
-            self.logger.info(f"Running inference for model BLIP Saffal")
-            BLIP_saffal_inference = self.__inference_saffal_blip(image, text)
-            return BLIP_saffal_inference
-
-        elif options == "Blip Control Net":
-            self.logger.info(f"Running inference for model BLIP Control Net")
-            BLIP_control_net_inference = self.__inference_control_net_blip(image, text)
-            return BLIP_control_net_inference
+    def inference(self, selected, image, text):
+        self.logger.info(f"selected model {selected}")
+        if selected == "Blip Saffal":
+            return self.__inference_saffal_blip(image, text)
+        elif selected == "Blip CN":
+            return self.__inference_control_net_blip(image, text)
+        else:
+            self.logger.warning("Please select a model to make the inference..")
 
     def __inference_saffal_blip(self, image, text):
         encoding = self.blip_processor(image, text, return_tensors="pt")