fix app

app.py

@@ -6,7 +6,8 @@ import gradio as gr
 from featup.util import norm, unnorm, pca, remove_axes
 from pytorch_lightning import seed_everything
 import os
-
+import requests
+import os
 
 def plot_feats(image, lr, hr):
     assert len(image.shape) == len(lr.shape) == len(hr.shape) == 3
@@ -40,9 +41,6 @@ def plot_feats(image, lr, hr):
 
 
 if __name__ == "__main__":
-    import requests
-    import os
-
 
     def download_image(url, save_path):
         response = requests.get(url)
@@ -56,7 +54,7 @@ if __name__ == "__main__":
         "plant.png": base_url + "plant.png",
     }
 
-    sample_images_dir = "sample_images"
+    sample_images_dir = "/tmp/sample_images"
 
     # Ensure the directory for sample images exists
     os.makedirs(sample_images_dir, exist_ok=True)
@@ -112,9 +110,9 @@ if __name__ == "__main__":
                         title="Feature Upsampling Demo",
                         description="This demo allows you to upsample features of an image using selected models.",
                         examples=[
-                            ["sample_images/skate.jpg", "dino16"],
-                            ["sample_images/car.jpg", "dinov2"],
-                            ["sample_images/plant.png", "dino16"],
+                            ["/tmp/sample_images/skate.jpg", "dino16"],
+                            ["/tmp/sample_images/car.jpg", "dinov2"],
+                            ["/tmp/sample_images/plant.png", "dino16"],
                         ]
 
                         )