pragnakalp commited on
Commit
55844f6
·
1 Parent(s): 889843a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -105,6 +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 '/content/image_pre.png'
109
 
110
 
@@ -138,8 +139,11 @@ def calculate(image_in, audio_in):
138
 
139
  def one_shot_talking(image_in,audio_in):
140
 
 
 
 
141
  #Pre-processing of image
142
- crop_src_image(image_in)
143
 
144
  exit()
145
  #Improve quality of input image
 
105
  else:
106
  img = cv2.resize(img,(256,256))
107
  cv2.imwrite(save_img, img)
108
+
109
  return '/content/image_pre.png'
110
 
111
 
 
139
 
140
  def one_shot_talking(image_in,audio_in):
141
 
142
+ image = Image.open(image_in)
143
+ image = crop_src_image(image)
144
+ image.save("image_pre.png")
145
  #Pre-processing of image
146
+ # crop_src_image(image_in)
147
 
148
  exit()
149
  #Improve quality of input image