Spaces:
Runtime error
Runtime error
Commit
•
f2f0545
1
Parent(s):
655cda8
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,9 @@ from PIL import Image
|
|
| 8 |
import gradio as gr
|
| 9 |
import os
|
| 10 |
|
|
|
|
|
|
|
|
|
|
| 11 |
#Load Magenta Arbitrary Image Stylization network
|
| 12 |
hub_module = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/1')
|
| 13 |
"""
|
|
|
|
| 8 |
import gradio as gr
|
| 9 |
import os
|
| 10 |
|
| 11 |
+
import os
|
| 12 |
+
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
|
| 13 |
+
|
| 14 |
#Load Magenta Arbitrary Image Stylization network
|
| 15 |
hub_module = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/1')
|
| 16 |
"""
|