Spaces:
Running
Running
Update app.py
#2
by
phuongpropk05
- opened
app.py
CHANGED
|
@@ -45,16 +45,16 @@ if pipeline_available:
|
|
| 45 |
with st.spinner("Generating video... This may take a while."):
|
| 46 |
try:
|
| 47 |
# Attempt to generate the video
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
|
|
|
|
| 45 |
with st.spinner("Generating video... This may take a while."):
|
| 46 |
try:
|
| 47 |
# Attempt to generate the video
|
| 48 |
+
video_frames = pipe(
|
| 49 |
+
prompt=prompt,
|
| 50 |
+
image=image,
|
| 51 |
+
num_videos_per_prompt=1,
|
| 52 |
+
num_inference_steps=50,
|
| 53 |
+
num_frames=81,
|
| 54 |
+
guidance_scale=6,
|
| 55 |
+
generator=torch.Generator(device=device).manual_seed(42),
|
| 56 |
+
).frames[0]
|
| 57 |
+
except Exception as e:
|
| 58 |
+
# Handle errors gracefully
|
| 59 |
+
st.error(f"An error occurred during video generation: {e}")
|
| 60 |
|