added spaces

app.py

@@ -7,7 +7,7 @@ import torch
 from PIL import Image
 import os
 import requests
-
+import spaces
 
 def get_config(hf_model_path):
     config_path = hf_hub_download(hf_model_path, "config.json")
@@ -57,7 +57,7 @@ SYSTEM_PROMPT = "You are an AI visual assistant and you are seeing a single imag
 whiteboard = Image.open(requests.get("https://c1.staticflickr.com/7/6168/6207108414_a8833f410e_o.jpg", stream=True).raw).convert('RGB')
 taxi_image = Image.open(requests.get("https://llava.hliu.cc/file=/nobackup/haotian/tmp/gradio/ca10383cc943e99941ecffdc4d34c51afb2da472/extreme_ironing.jpg", stream=True).raw).convert('RGB')
 
-
+@spaces.GPU
 def generate_answer(img, question, max_new_tokens, num_beams):
     
     image = [img]