DHEIVER commited on
Commit
8e9f292
·
1 Parent(s): d0315e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -74,13 +74,21 @@ class ImageClassifierCataract:
74
  </div>
75
  """
76
 
77
- return output_image, predicted_class, current_time.strftime('%Y-%m-%d %H:%M:%S'), info_html
 
 
 
 
 
 
 
 
78
 
79
  def run_interface(self):
80
  input_interface = gr.Interface(
81
  fn=self.classify_image,
82
  inputs="image",
83
- outputs=["image", "text", "text", "html"],
84
  live=True
85
  )
86
  input_interface.launch()
 
74
  </div>
75
  """
76
 
77
+ # Combine the output elements
78
+ output = {
79
+ "image": output_image,
80
+ "text": predicted_class,
81
+ "text_1": current_time.strftime('%Y-%m-%d %H:%M:%S'),
82
+ "html": info_html,
83
+ }
84
+
85
+ return output
86
 
87
  def run_interface(self):
88
  input_interface = gr.Interface(
89
  fn=self.classify_image,
90
  inputs="image",
91
+ outputs=["image", "text", "text_1", "html"],
92
  live=True
93
  )
94
  input_interface.launch()