Update app.py

app.py

@@ -45,19 +45,42 @@ ci = Interrogator(config)
 def inference(image, mode, best_max_flavors):
     image = image.convert('RGB')
     if mode == 'best':
-        print("mode: best")
+        
+        print("""————
+        new inference
+            mode: best
+        """)
+        
         prompt_result = ci.interrogate(image, max_flavors=int(best_max_flavors))
+        
         print(prompt_result)
+        
         return prompt_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    
     elif mode == 'classic':
-        print("mode: classic")
+        
+        print("""————
+        new inference
+            mode: classic
+        """)
+        
         prompt_result = ci.interrogate_classic(image)
+        
         print(prompt_result)
+        
         return prompt_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    
     else:
-        print("mode: fast")
+        
+        print("""————
+        new inference
+            mode: best
+        """)
+        
         prompt_result = ci.interrogate_fast(image)
+        
         print(prompt_result)
+        
         return prompt_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 title = """