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)
|