Add spaces GPU

app.py

@@ -2,6 +2,7 @@ from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
 from prompt import smoke_detection_prompt
 import gradio as gr
+import spaces
 
 model_name = "leon-se/ForestFireVLM-3B"
 model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
@@ -9,6 +10,7 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 )
 processor = AutoProcessor.from_pretrained(model_name)
 
+@spaces.GPU(duration=120)
 def generate(image):
     messages = [
         {

requirements.txt

@@ -3,4 +3,5 @@ transformers
 gradio
 torchvision
 torch
-accelerate
+accelerate
+spaces