Update app.py

app.py

@@ -6,7 +6,6 @@ import spaces
 import sys
 import torch
 from shap_e.models.transmitter.base import Transmitter, VectorDecoder
-from shap_e.util.collections import AttrDict
 
 import torch
 import torch.nn as nn
@@ -539,7 +538,7 @@ def create_demo():
             )
 
         # Set up event handlers
-        @spaces.GPU(duration=20)  # Add GPU decorator to the generate function
+        @spaces.GPU(duration=80)  # Add GPU decorator to the generate function
         def generate(prompt, guidance_scale, num_steps):
             try:
                 torch.cuda.empty_cache()  # Clear GPU memory before starting