mhamilton723 commited on
Commit
523ffdf
1 Parent(s): e6c364b

fix csv issue

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,6 +8,7 @@ from pytorch_lightning import seed_everything
8
  import os
9
  import requests
10
  import os
 
11
 
12
  def plot_feats(image, lr, hr):
13
  assert len(image.shape) == len(lr.shape) == len(hr.shape) == 3
@@ -71,7 +72,7 @@ if __name__ == "__main__":
71
 
72
  os.environ['TORCH_HOME'] = '/tmp/.cache'
73
  os.environ['GRADIO_EXAMPLES_CACHE'] = '/tmp/gradio_cache'
74
-
75
  options = ['dino16', 'vit', 'dinov2', 'clip', 'resnet50']
76
 
77
  image_input = gr.Image(label="Choose an image to featurize",
 
8
  import os
9
  import requests
10
  import os
11
+ import csv
12
 
13
  def plot_feats(image, lr, hr):
14
  assert len(image.shape) == len(lr.shape) == len(hr.shape) == 3
 
72
 
73
  os.environ['TORCH_HOME'] = '/tmp/.cache'
74
  os.environ['GRADIO_EXAMPLES_CACHE'] = '/tmp/gradio_cache'
75
+ csv.field_size_limit(100000000)
76
  options = ['dino16', 'vit', 'dinov2', 'clip', 'resnet50']
77
 
78
  image_input = gr.Image(label="Choose an image to featurize",