pragnakalp commited on
Commit
889843a
·
1 Parent(s): 2e0398b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -105,7 +105,7 @@ def crop_src_image(src_img, detector=None):
105
  else:
106
  img = cv2.resize(img,(256,256))
107
  cv2.imwrite(save_img, img)
108
- return save_img
109
 
110
 
111
  def pad_image(image):
@@ -173,8 +173,6 @@ def one_shot(image,input_text,gender):
173
  f.seek(0)
174
  sound = AudioSegment.from_file(f.name, format="mp3")
175
  sound.export("/content/audio.wav", format="wav")
176
- return "image"
177
- exit()
178
 
179
  one_shot_talking(image,'/content/audio.wav')
180
 
 
105
  else:
106
  img = cv2.resize(img,(256,256))
107
  cv2.imwrite(save_img, img)
108
+ return '/content/image_pre.png'
109
 
110
 
111
  def pad_image(image):
 
173
  f.seek(0)
174
  sound = AudioSegment.from_file(f.name, format="mp3")
175
  sound.export("/content/audio.wav", format="wav")
 
 
176
 
177
  one_shot_talking(image,'/content/audio.wav')
178