add release GPU decoration for generate fn

app.py

@@ -1,3 +1,4 @@
+import spaces
 from fastapi import FastAPI, HTTPException, UploadFile, File
 from typing import Optional, Dict, Any
 import torch
@@ -338,6 +339,7 @@ def create_gradio_interface():
                 )
             ]
         
+        @spaces.GPU
         def generate(model_name, prompt_text, h, w, steps, guide_scale, ref_img):
             response = generate_image(
                 model_name=model_name,

requirements.txt

@@ -26,13 +26,10 @@
 --extra-index-url https://download.pytorch.org/whl/cu113
 fastapi==0.95.2
 uvicorn==0.22.0
-
-torchvision==0.13.1+cu113
-torchaudio==0.12.1+cu113
 diffusers
 transformers==4.26.1
 Pillow==9.4.0
 numpy==1.23.5
 gradio==3.16.2
 huggingface_hub
-torch
+torch==2.4.0