Update app.py

app.py

@@ -40,7 +40,7 @@ def check_safety(text):
     return "Unsafe" if pred[0] == 1 else "Safe"
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=60)
 def inference(prompt: str, max_tokens: int = 256) -> str:
     model.to("cuda")  # Move to GPU inside decorated function