Update for ZeroGPU

app.py

@@ -8,6 +8,7 @@ import random
 import gradio as gr
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 from diffusers import KandinskyV22Pipeline, KandinskyV22PriorPipeline
 
@@ -42,6 +43,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
+@spaces.GPU
 def generate(
     prompt: str,
     negative_prompt: str = "low quality, bad quality",

requirements.txt

@@ -1,6 +1,6 @@
 accelerate==0.23.0
 diffusers==0.21.2
 gradio==3.44.4
-torch==2.0.1
-torchvision==0.15.2
+torch==2.0.0
+torchvision==0.15.1
 transformers==4.33.2