update

app.py

@@ -26,6 +26,7 @@ import collections
 import numpy as np
 import gradio as gr
 from PIL import Image
+import spaces
 
 import torch
 from pathlib import Path
@@ -131,6 +132,7 @@ def get_points(img, original_image, sel_pix, evt: gr.SelectData):
     return [img if isinstance(img, np.ndarray) else np.array(img), original_image]
 
 
+@spaces.GPU(duration=120)
 def gesturenet_inference(ref_image, prompt, selected_points):
 
     # Check some paramter, must have prompt and selected points
@@ -178,7 +180,7 @@ def gesturenet_inference(ref_image, prompt, selected_points):
         project_config = ProjectConfiguration(project_dir=config["output_dir"], logging_dir=Path(config["output_dir"], config["logging_name"])),
     )
     print("device is ", accelerator.device)
-    
+
     feature_extractor = CLIPImageProcessor.from_pretrained(
         config["pretrained_model_name_or_path"], subfolder="feature_extractor", revision=None
     )   # This instance has now weight, they are just seeting file

requirements.txt

@@ -1,7 +1,7 @@
 # Non-strict version lib
-# torch==2.0.1
-# torchaudio==2.0.1
-# torchvision
+torch==2.0.1
+torchaudio==2.0.1
+torchvision
 opencv-python
 transformers
 accelerate