Update app.py

app.py

@@ -139,11 +139,15 @@ def one_shot_talking(image_in,audio_in):
   crop_img=crop_src_image(image_in)
 
   # os.system(f"rm -rf /content/results/restored_imgs/image_pre.png")
+  
+  if not os.path.exists( "/content/results" ):
+    os.makedirs("/content/results")
+  
   #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")
   # time.sleep(60)
   
-  image_in_one_shot='/content/results/restored_imgs/image_pre.png'
+  image_in_one_shot='/content/results/image_pre.png'
   
   #One Shot Talking Face algorithm
   calculate(image_in_one_shot,audio_in)