Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,9 +10,10 @@ pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dt
|
|
10 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
11 |
pipe.enable_model_cpu_offload()
|
12 |
|
13 |
-
|
14 |
|
15 |
def create_image_caption(image_init):
|
|
|
16 |
cap = caption(image_init, "Nucleus sampling", 1.2, 0.5, 5, 20, fn_index=0)
|
17 |
print("cap: " + cap)
|
18 |
return cap
|
|
|
10 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
11 |
pipe.enable_model_cpu_offload()
|
12 |
|
13 |
+
|
14 |
|
15 |
def create_image_caption(image_init):
|
16 |
+
caption = gr.load(name="spaces/fffiloni/CoCa-clone")
|
17 |
cap = caption(image_init, "Nucleus sampling", 1.2, 0.5, 5, 20, fn_index=0)
|
18 |
print("cap: " + cap)
|
19 |
return cap
|