Spaces:
Running
on
T4
Running
on
T4
Ahsen Khaliq
commited on
Commit
•
d08bf1f
1
Parent(s):
d4ad940
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,6 @@ import torch
|
|
| 4 |
import gradio as gr
|
| 5 |
os.system("pip install gradio==2.5.3")
|
| 6 |
|
| 7 |
-
|
| 8 |
model2 = torch.hub.load(
|
| 9 |
"AK391/animegan2-pytorch:main",
|
| 10 |
"generator",
|
|
@@ -12,6 +11,7 @@ model2 = torch.hub.load(
|
|
| 12 |
device="cuda",
|
| 13 |
progress=False
|
| 14 |
)
|
|
|
|
| 15 |
model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
|
| 16 |
face2paint = torch.hub.load(
|
| 17 |
'AK391/animegan2-pytorch:main', 'face2paint',
|
| 4 |
import gradio as gr
|
| 5 |
os.system("pip install gradio==2.5.3")
|
| 6 |
|
|
|
|
| 7 |
model2 = torch.hub.load(
|
| 8 |
"AK391/animegan2-pytorch:main",
|
| 9 |
"generator",
|
| 11 |
device="cuda",
|
| 12 |
progress=False
|
| 13 |
)
|
| 14 |
+
|
| 15 |
model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1", device="cuda")
|
| 16 |
face2paint = torch.hub.load(
|
| 17 |
'AK391/animegan2-pytorch:main', 'face2paint',
|