Update for ZeroGPU

README.md

@@ -5,7 +5,7 @@ colorFrom: blue
 colorTo: pink
 sdk: gradio
 sdk_version: 3.44.4
-python_version: 3.10.12
+python_version: 3.10.11
 app_file: app.py
 pinned: false
 license: bsd-3-clause

app.py

@@ -7,6 +7,7 @@ import string
 
 import gradio as gr
 import PIL.Image
+import spaces
 import torch
 from transformers import AutoProcessor, Blip2ForConditionalGeneration
 
@@ -29,6 +30,7 @@ else:
     model = None
 
 
+@spaces.GPU
 def generate_caption(
     image: PIL.Image.Image,
     decoding_method: str,
@@ -52,6 +54,7 @@ def generate_caption(
     return result
 
 
+@spaces.GPU
 def answer_question(
     image: PIL.Image.Image,
     text: str,

requirements.txt

@@ -4,6 +4,6 @@ gradio==3.44.4
 huggingface-hub==0.17.2
 Pillow==10.0.1
 scipy==1.11.2
-torch==2.0.1
-torchvision==0.15.2
+torch==2.0.0
+torchvision==0.15.1
 transformers==4.33.2