Update app

app.py

@@ -4,6 +4,7 @@ from io import BytesIO
 from pathlib import Path
 
 import glob
+import spaces
 import numpy as np
 import gradio as gr
 import rasterio as rio
@@ -164,6 +165,7 @@ def _bar_chart(top_scores, img_name, cmap) -> str:
     return f'<img src="data:image/png;base64,{b64}" style="display:block;margin:auto;width:{IMG_PX}px;" />'
 
 
+@spaces.GPU
 def classify(images, class_text):
     class_names = [c.strip() for c in class_text.split(",") if c.strip()]
     cards = []

requirements.txt

@@ -1,3 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
 gradio>=4.31.0
 plotly
 rasterio