Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import wget
|
|
6 |
import torch
|
7 |
import gradio
|
8 |
|
9 |
-
os.system("git clone https://github.com/NVlabs/stylegan3")
|
10 |
sys.path.append('./stylegan3')
|
11 |
|
12 |
model_names = {
|
@@ -44,7 +44,7 @@ def load_model(file_name: str, device: torch.device):
|
|
44 |
|
45 |
local_path = '/content/'f'{file_name}'
|
46 |
print(local_path)
|
47 |
-
with open(fetch_model(
|
48 |
model = pickle.load(f)['G_ema']
|
49 |
model.eval()
|
50 |
model.to(device)
|
|
|
6 |
import torch
|
7 |
import gradio
|
8 |
|
9 |
+
#os.system("git clone https://github.com/NVlabs/stylegan3")
|
10 |
sys.path.append('./stylegan3')
|
11 |
|
12 |
model_names = {
|
|
|
44 |
|
45 |
local_path = '/content/'f'{file_name}'
|
46 |
print(local_path)
|
47 |
+
with open(fetch_model(local_path), 'rb') as f:
|
48 |
model = pickle.load(f)['G_ema']
|
49 |
model.eval()
|
50 |
model.to(device)
|