Spaces:
Runtime error
Runtime error
Commit
•
523ffdf
1
Parent(s):
e6c364b
fix csv issue
Browse files
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",
|