jcmc commited on
Commit
358206b
·
1 Parent(s): 97adf14

adjustments

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -266,14 +266,14 @@ def inference(
266
  video_writer = video_writer
267
  )
268
 
269
- out.save(f'dip_{timestring}.png', quality=100)
270
  if save_progress_video:
271
  video_writer.close()
272
- return out, 'video.mp4'
273
 
274
  iface = gr.Interface(fn=inference,
275
  inputs=["number", "number", "number", "number", "number",
276
  gr.inputs.Checkbox(default=False, label="class_token"),
277
  gr.inputs.Checkbox(default=True, label="maximise"),
278
  "number"],
279
- outputs=["image","video"]).launch()
 
266
  video_writer = video_writer
267
  )
268
 
269
+ # out.save(f'dip_{timestring}.png', quality=100)
270
  if save_progress_video:
271
  video_writer.close()
272
+ return out
273
 
274
  iface = gr.Interface(fn=inference,
275
  inputs=["number", "number", "number", "number", "number",
276
  gr.inputs.Checkbox(default=False, label="class_token"),
277
  gr.inputs.Checkbox(default=True, label="maximise"),
278
  "number"],
279
+ outputs=["image","video"]).launch(enable_queue=True,cache_examples=True)