Update app.py

app.py

@@ -10,11 +10,11 @@ import spaces
 extensions_dir = "./torch_extension/"
 os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
 
-from networks.generator import Generator
-
-
+@spaces.GPU
 def generator(prompt):
 
+    from networks.generator import Generator
+    
     return prompt
 
 gr.Interface(