Update app.py

app.py

@@ -105,7 +105,7 @@ def crop_src_image(src_img, detector=None):
     else:
         img = cv2.resize(img,(256,256))
         cv2.imwrite(save_img, img)
-
+    return save_img
 
 
 def pad_image(image):
@@ -141,6 +141,7 @@ def one_shot_talking(image_in,audio_in):
   #Pre-processing of image
   crop_src_image(image_in)
 
+  exit()
   #Improve quality of input image
   os.system(f"python /content/GFPGAN/inference_gfpgan.py --upscale 2 -i /content/image_pre.png -o /content/results  --bg_upsampler realesrgan")
 
@@ -161,7 +162,7 @@ def one_shot_talking(image_in,audio_in):
   merge_frames()
 
   audio_video()
-  return "Sucessufull"
+  
 
     
 def one_shot(image,input_text,gender): 
@@ -214,7 +215,7 @@ def run():
           # audio_in = gr.Audio(show_label=False, type='filepath')
           input_text=gr.Textbox(lines=3, value="Hello How are you?", label="Input Text")
           gender = gr.Radio(["Female","Male"],value="Female",label="Gender")
-          video_out = gr.Audio(label="output")
+          video_out = gr.Image(label="output")
           # video_out = gr.Video(show_label=False)
         with gr.Row().style(equal_height=True):
           btn = gr.Button("Generate")