fix examples, autocast, add spaces dependency

app.py

@@ -4,7 +4,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 import torchvision.transforms.functional as TF
 import math
-import random
+import spaces
 import numpy as np
 from PIL import Image
 
@@ -197,6 +197,7 @@ print(f"Best val PSNR: {checkpoint['val_psnr']:.2f} dB")
 
 
 # ── Inference ─────────────────────────────────────────────────────────────────
+@spaces.GPU
 def run_sr(img_pil):
     """
     Takes any PIL image.
@@ -264,11 +265,11 @@ def run_sr(img_pil):
 # smooth skin + sharp boundary, urban grid, natural foliage
 EXAMPLES = [
     ["examples/urban.png"],
-    ["examples/satellite.png"],
-    ["examples/document.png"],
-    ["examples/portrait.png"],
+    ["examples/aerial.png"],
     ["examples/architecture.png"],
     ["examples/nature.png"],
+    ["examples/portrait.png"],
+    ["examples/texture.png"],
 ]
 
 
@@ -470,7 +471,6 @@ with gr.Blocks(css=CSS, title="Dense-Iso-ViT SR") as demo:
         examples=EXAMPLES,
         inputs=[input_img],
         label="Examples — showing V4 strengths",
-        elem_classes=["examples-holder"],
     )
 
     # collapsible architecture info

requirements.txt

@@ -3,3 +3,4 @@ torchvision>=0.15.0
 Pillow>=9.0.0
 numpy>=1.24.0
 huggingface_hub<1.0
+spaces