Update app.py

app.py

@@ -18,7 +18,7 @@ def encode_prompt(prompt: str):
         scheduler=None,          # inutile
         safety_checker=None,     # inutile
         feature_extractor=None,  # inutile
-    )
+    ).to("cpu")
 
     with torch.inference_mode():
         prompt_embeds = pipe.encode_prompt(prompt)