Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ model_vaq_g = GumbelVQ(
|
|
58 |
transform = T.Compose([
|
59 |
T.Resize((256, 256)),
|
60 |
T.ToTensor(),
|
61 |
-
T.Normalize(mean=[0, 0, 0], std=[0.
|
62 |
|
63 |
#_________________Define:Gradio Function________________________
|
64 |
|
|
|
58 |
transform = T.Compose([
|
59 |
T.Resize((256, 256)),
|
60 |
T.ToTensor(),
|
61 |
+
T.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])]) # Normalize to [-1, 1]
|
62 |
|
63 |
#_________________Define:Gradio Function________________________
|
64 |
|