Update app.py

app.py

@@ -45,7 +45,7 @@ def encode_text(prompt: str):
     # Sauvegarde
     fd, path = tempfile.mkstemp(suffix=".pt")
     os.close(fd)
-    torch.save(outputs.cpu(), path)
+    torch.save(outputs, path)
 
     return path