srkanth

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import spaces
 import os 
 from PIL import Image
 import torch
@@ -11,6 +10,7 @@ import argparse
 import numpy as np
 import torchvision.transforms.functional as TF
 from scipy.ndimage import convolve, zoom
+import spaces
 from utils import resize_by_short_side
 
 from tools.interact_tools import SamControler

requirements.txt

@@ -1,5 +1,5 @@
 accelerate
-torch==2.4.0
+torch==2.2.0
 torchvision
 numpy==1.26.4
 opencv-python