aakashch0179 commited on
Commit
26ce3ee
1 Parent(s): 89e3069

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -45,7 +45,8 @@ import streamlit as st
45
  import torch
46
  from diffusers import ShapEPipeline
47
  from diffusers.utils import export_to_gif
48
- from PIL import Image
 
49
 
50
  # Model loading (Ideally done once at the start for efficiency)
51
  ckpt_id = "openai/shap-e"
 
45
  import torch
46
  from diffusers import ShapEPipeline
47
  from diffusers.utils import export_to_gif
48
+ from PIL import Image
49
+ import numpy as np
50
 
51
  # Model loading (Ideally done once at the start for efficiency)
52
  ckpt_id = "openai/shap-e"