update space

app.py

@@ -5,7 +5,7 @@ set environment variable OPENAI_API_KEY to use the OpenAI API to enhance the pro
 Usage:
     OPENAI_API_KEY=your_openai_api_key OPENAI_BASE_URL=your_base_url python app.py
 """
-
+import spaces
 import math
 import os
 import random
@@ -210,7 +210,7 @@ def convert_prompt(prompt: str, retry_times: int = 3) -> str:
             return response.choices[0].message.content
     return prompt
 
-
+@spaces.GPU
 def infer(
     prompt: str,
     image_input: str,