will33am commited on
Commit
909e4ad
1 Parent(s): 1d0c146
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -13,7 +13,11 @@ from PIL import Image
13
 
14
  DICT_CLASSES = {'lake':955,
15
  'castle':483,
16
- 'library':624}
 
 
 
 
17
  IMG_MAX_SIZE = 256
18
  ARCH = 'crossvit_18_dagger_408'
19
  ARCH_PATH = './checkpoints/robust_crossvit_18_dagger_408.pt'
@@ -83,4 +87,6 @@ if __name__ == '__main__':
83
  calculate_button.click(fn = gradio_fn,
84
  inputs = [image_input,radio_steps,radio_class,radio_robust],
85
  outputs = target_image)
86
- demo.launch(debug = True)
 
 
 
13
 
14
  DICT_CLASSES = {'lake':955,
15
  'castle':483,
16
+ 'library':624,
17
+ 'dog':235,
18
+ 'cat':285,
19
+ 'people':842 #trunks
20
+ }
21
  IMG_MAX_SIZE = 256
22
  ARCH = 'crossvit_18_dagger_408'
23
  ARCH_PATH = './checkpoints/robust_crossvit_18_dagger_408.pt'
 
87
  calculate_button.click(fn = gradio_fn,
88
  inputs = [image_input,radio_steps,radio_class,radio_robust],
89
  outputs = target_image)
90
+ demo.launch(share = True,debug = True)
91
+
92
+
app.py CHANGED
@@ -13,7 +13,11 @@ from PIL import Image
13
 
14
  DICT_CLASSES = {'lake':955,
15
  'castle':483,
16
- 'library':624}
 
 
 
 
17
  IMG_MAX_SIZE = 256
18
  ARCH = 'crossvit_18_dagger_408'
19
  ARCH_PATH = './checkpoints/robust_crossvit_18_dagger_408.pt'
@@ -83,4 +87,6 @@ if __name__ == '__main__':
83
  calculate_button.click(fn = gradio_fn,
84
  inputs = [image_input,radio_steps,radio_class,radio_robust],
85
  outputs = target_image)
86
- demo.launch(debug = True)
 
 
 
13
 
14
  DICT_CLASSES = {'lake':955,
15
  'castle':483,
16
+ 'library':624,
17
+ 'dog':235,
18
+ 'cat':285,
19
+ 'people':842 #trunks
20
+ }
21
  IMG_MAX_SIZE = 256
22
  ARCH = 'crossvit_18_dagger_408'
23
  ARCH_PATH = './checkpoints/robust_crossvit_18_dagger_408.pt'
 
87
  calculate_button.click(fn = gradio_fn,
88
  inputs = [image_input,radio_steps,radio_class,radio_robust],
89
  outputs = target_image)
90
+ demo.launch(share = True,debug = True)
91
+
92
+
src/__pycache__/utils.cpython-39.pyc ADDED
Binary file (1.81 kB). View file