fffiloni commited on
Commit
f508df4
1 Parent(s): 73bd937

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- caption = gr.load(name="spaces/fffiloni/CoCa-clone")
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