Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import (
     __version__,
     GenerationConfig,
 )
+import spaces
 from PIL import Image
 import gradio as gr
 import argparse
@@ -56,7 +57,7 @@ generation_config = GenerationConfig.from_dict(
     }
 )
 
-
+@spaces.GPU
 def generate(image: Image):
     with tempfile.NamedTemporaryFile(suffix=".jpg", delete=True) as tmp:
         image.save(tmp.name)