Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -88,7 +88,7 @@ def main():
|
|
88 |
log_visual_freq=1000,
|
89 |
input='text',
|
90 |
))
|
91 |
-
|
92 |
#generator = create_generator(args, device)
|
93 |
iface = gr.Interface(
|
94 |
fn=predict,
|
|
|
88 |
log_visual_freq=1000,
|
89 |
input='text',
|
90 |
))
|
91 |
+
device = th.device()
|
92 |
#generator = create_generator(args, device)
|
93 |
iface = gr.Interface(
|
94 |
fn=predict,
|