iamrobotbear commited on
Commit
e7aada3
·
1 Parent(s): e75b52b

almost there i think

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from gradio import components #add for csv?
3
  import torch
4
  from PIL import Image
5
  import pandas as pd
@@ -12,6 +12,7 @@ import io
12
  from sklearn.metrics.pairwise import cosine_similarity
13
 
14
 
 
15
  # Import logging module
16
  import logging
17
 
@@ -137,9 +138,9 @@ def process_images_and_statements(image):
137
  return results_df, csv_results # <--- Return results_df and csv_results
138
 
139
  # Gradio interface
140
- image_input = gr.inputs.Image()
141
- output_df = gr.outputs.Dataframe(type="pandas", label="Results")
142
- output_csv = gr.outputs.File(label="Download CSV")
143
 
144
  iface = gr.Interface(
145
  fn=process_images_and_statements,
 
1
  import gradio as gr
2
+ from gradio import inputs, outputs # Import inputs and outputs from gradio
3
  import torch
4
  from PIL import Image
5
  import pandas as pd
 
12
  from sklearn.metrics.pairwise import cosine_similarity
13
 
14
 
15
+
16
  # Import logging module
17
  import logging
18
 
 
138
  return results_df, csv_results # <--- Return results_df and csv_results
139
 
140
  # Gradio interface
141
+ image_input = inputs.Image() # Use inputs instead of gr.inputs
142
+ output_df = outputs.Dataframe(type="pandas", label="Results") # Use outputs instead of gr.outputs
143
+ output_csv = outputs.File(label="Download CSV") # Use outputs instead of gr.outputs
144
 
145
  iface = gr.Interface(
146
  fn=process_images_and_statements,