Update app.py

app.py

@@ -13,7 +13,7 @@ class StopOnTokens(StoppingCriteria):
                 return True
         return False
 
-@spaces.GPU
+@spaces.GPU(duration=480)
 def predict(message, history):
     torch.set_default_device("cuda")
 
@@ -54,7 +54,4 @@ def predict(message, history):
             break
         yield partial_message
 
-
-gr.ChatInterface(predict,
-    theme=gr.themes.Soft(primary_hue="purple"),
-).launch()
+gr.ChatInterface(predict).launch()