Update app.py

app.py

@@ -119,7 +119,8 @@ from PIL import Image
 import numpy as np 
 
 # Model loading 
-ckpt_id = "openai/shap-e"  
+ckpt_id = "openai/shap-e"
+StableDiffusionPipeline.from_pretrained(ckpt_id, torch_dtype=torch.float16, use_auth_token=True).to("cuda") 
 @st.cache_resource
 def load_model():
     return StableDiffusionPipeline.from_pretrained(ckpt_id).to("cuda")