Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|