Update app.py

app.py

@@ -45,6 +45,8 @@ def load_language_model():
     return language_tokenizer, language_model
 
 # 从图片生成描述
+# 定义生成响应的函数，并使用 @spaces.GPU 装饰
+@spaces.GPU(duration=40)  # 建议将 duration 增加到 120
 def generate_image_description(image):
     vqa_processor, vqa_model = load_vqa_model()
     conversation = [
@@ -65,6 +67,8 @@ def generate_image_description(image):
     return image_description
 
 # 使用纯语言模型生成最终回答
+# 定义生成响应的函数，并使用 @spaces.GPU 装饰
+@spaces.GPU(duration=40)  # 建议将 duration 增加到 120
 def generate_language_response(instruction, image_description):
     language_tokenizer, language_model = load_language_model()
     prompt = f"""### Instruction: