Hugging Face's logo

Spaces:
LPX55
/
QwenStorytellerV2
Running on Zero

App Files Community
1
LPX55 commited on
Commit
84a7aed
·
verified ·
1 Parent(s): 7783d7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
26
  )
27
  processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller2")
28
 
29
- @spaces.GPU()
30
  @torch.no_grad()
31
  def generate_story(file_paths, progress=gr.Progress(track_tqdm=True)):
32
  # Load images from the file paths
 
26
  )
27
  processor = AutoProcessor.from_pretrained("daniel3303/QwenStoryteller2")
28
 
29
+ @spaces.GPU(duration=180)
30
  @torch.no_grad()
31
  def generate_story(file_paths, progress=gr.Progress(track_tqdm=True)):
32
  # Load images from the file paths