taesiri commited on
Commit
2731d0d
1 Parent(s): fb992f3

update to blocks

Browse files
Files changed (2) hide show
  1. app.py +21 -5
  2. requirements.txt +6 -6
app.py CHANGED
@@ -125,14 +125,14 @@ def visualize_similarities(image1, image2):
125
 
126
 
127
  # GRADIO APP
128
- iface = gr.Interface(
129
  fn=visualize_similarities,
130
  inputs=[
131
- gr.inputs.Image(type="pil"),
132
- gr.inputs.Image(type="pil"),
133
  ],
134
  allow_flagging="never",
135
- outputs=[gr.Plot(type="matplotlib")],
136
  cache_examples=True,
137
  enable_queue=False,
138
  examples=[
@@ -143,4 +143,20 @@ iface = gr.Interface(
143
  ],
144
  )
145
 
146
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
125
 
126
 
127
  # GRADIO APP
128
+ main = gr.Interface(
129
  fn=visualize_similarities,
130
  inputs=[
131
+ gr.Image(type="pil"),
132
+ gr.Image(type="pil"),
133
  ],
134
  allow_flagging="never",
135
+ outputs=gr.Plot(type="matplotlib"),
136
  cache_examples=True,
137
  enable_queue=False,
138
  examples=[
143
  ],
144
  )
145
 
146
+ # iface.launch()
147
+
148
+ blocks = gr.Blocks()
149
+ with blocks:
150
+
151
+ gr.Markdown(
152
+ """
153
+ # Visualizing Deep Similarity Networks
154
+ A quick demo to visualize the similarity between two images.
155
+ [Original Paper](https://arxiv.org/pdf/1901.00536.pdf) - [Github Page](https://github.com/GWUvision/Similarity-Visualization)
156
+ """
157
+ )
158
+
159
+ gr.TabbedInterface([main], ["Main"])
160
+
161
+
162
+ blocks.launch(debug=True)
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  gradio==3.0.5
2
- matplotlib==3.4.3
3
- numpy==1.21.2
4
- Pillow==8.4.0
5
- scikit-image==0.18.3
6
- torch==1.10.0
7
- torchvision==0.11.1
1
  gradio==3.0.5
2
+ matplotlib==3.5.2
3
+ numpy==1.21.5
4
+ Pillow==9.0.1
5
+ scikit-image==0.19.2
6
+ torch==1.11.0
7
+ torchvision==0.12.0