Ahsen Khaliq commited on
Commit
0699239
1 Parent(s): 7186525

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -1,7 +1,14 @@
1
  import os
2
  import gradio as gr
3
- os.system("gdown https://drive.google.com/uc?id=1BnS1T39GQr6LH0J5F1EXliIhgZhJ7EGl")
4
- os.system("gdown https://drive.google.com/uc?id=1PedYtI-K26FIpr9Iq2xrEgT9_DnPzble")
 
 
 
 
 
 
 
5
  from PIL import Image
6
 
7
  def inference(content, style):
 
1
  import os
2
  import gradio as gr
3
+ os.system("pip install googledrivedownloader")
4
+ from google_drive_downloader import GoogleDriveDownloader as gdd
5
+ gdd.download_file_from_google_drive(file_id='1BnS1T39GQr6LH0J5F1EXliIhgZhJ7EGl',
6
+ dest_path='./blendgan.pt',
7
+ unzip=False)
8
+ gdd.download_file_from_google_drive(file_id='1PedYtI-K26FIpr9Iq2xrEgT9_DnPzble',
9
+ dest_path='./psp_encoder.pt',
10
+ unzip=False)
11
+
12
  from PIL import Image
13
 
14
  def inference(content, style):