Fabrice-TIERCELIN commited on
Commit
4bcd590
·
verified ·
1 Parent(s): f9163ba
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -92,13 +92,13 @@ for i in range(3):
92
 
93
  print("Optimizing pipeline...")
94
 
95
- #optimize_pipeline_(pipe,
96
- # image=Image.new('RGB', (MAX_DIMENSION, MIN_DIMENSION)),
97
- # prompt='prompt',
98
- # height=MIN_DIMENSION,
99
- # width=MAX_DIMENSION,
100
- # num_frames=MAX_FRAMES_MODEL,
101
- #)
102
  print("All models loaded and optimized. Gradio app is ready.")
103
 
104
 
 
92
 
93
  print("Optimizing pipeline...")
94
 
95
+ optimize_pipeline_(pipe,
96
+ image=Image.new('RGB', (MAX_DIMENSION, MIN_DIMENSION)),
97
+ prompt='prompt',
98
+ height=MIN_DIMENSION,
99
+ width=MAX_DIMENSION,
100
+ num_frames=MAX_FRAMES_MODEL,
101
+ )
102
  print("All models loaded and optimized. Gradio app is ready.")
103
 
104