TheTrice3 commited on
Commit
bf4cc6c
·
verified ·
1 Parent(s): 64eaab1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -175,7 +175,7 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
175
  ).to(device).eval()
176
 
177
 
178
- @spaces.GPU
179
  def generate_image(model_name: str, text: str, image: Image.Image,
180
  max_new_tokens: int, temperature: float, top_p: float,
181
  top_k: int, repetition_penalty: float):
 
175
  ).to(device).eval()
176
 
177
 
178
+ @spaces.GPU(duration=600)
179
  def generate_image(model_name: str, text: str, image: Image.Image,
180
  max_new_tokens: int, temperature: float, top_p: float,
181
  top_k: int, repetition_penalty: float):