Omnibus commited on
Commit
25f4a9c
·
1 Parent(s): 7839840

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -241,12 +241,17 @@ def check_load(inp_url,outp_vid,hid_box,start_f,end_f):
241
  return out_trim,in_vid,trim_count
242
  #outp_vid.change(echo_fn,outp_vid,[out_trim])
243
  def process_image_1(image):
 
244
  #out = esr(img=image, model_name="realesr-general-x4v3", denoise_strength=0.5, face_enhance=False, outscale=2)
245
- out = esr([image, "realesr-general-x4v3", "0.5", "False", "2"])
 
 
 
246
  print (out)
247
  out = Image.open(out)
248
  return out
249
  def improve_quality():
 
250
  clip1 = VideoFileClip(f"{uid}-clip.mp4")
251
  clip = clip1.fl_image(process_image_1)
252
  clip.write_videofile(f"{uid}-clip-high.mp4")
 
241
  return out_trim,in_vid,trim_count
242
  #outp_vid.change(echo_fn,outp_vid,[out_trim])
243
  def process_image_1(image):
244
+ rand_im = uuid.uuid4()
245
  #out = esr(img=image, model_name="realesr-general-x4v3", denoise_strength=0.5, face_enhance=False, outscale=2)
246
+ cv2.imwrite(f"{rand_im}-vid_tmp_proc.png", frame_f)
247
+ out = os.path.abspath(f"{rand_im}-vid_tmp_proc.png")
248
+ out_url = f'https://omnibus-vid-url-dl-mod.hf.space/file={out}'
249
+ out = esr([out_url, "realesr-general-x4v3", "0.5", "False", "2"])
250
  print (out)
251
  out = Image.open(out)
252
  return out
253
  def improve_quality():
254
+
255
  clip1 = VideoFileClip(f"{uid}-clip.mp4")
256
  clip = clip1.fl_image(process_image_1)
257
  clip.write_videofile(f"{uid}-clip-high.mp4")