Spaces:

himanshu-skid19
/

Unconditional_Image_Generation_Using_Diffusion_Models

Sleeping

App Files Files Community

himanshu-skid19 commited on Oct 10, 2023

Commit

9d5daa8

·

1 Parent(s): cb95512

Update app.py

i think i fixed the deserialization issue now

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -184,7 +184,7 @@ def sample(model, image_size, batch_size=16, channels=3):
 model = SimpleUnet()
 st.title("Generatig images using a diffusion model")
-model.load_state_dict(torch.load("new_linear_model_1090.pt"), map_location=torch.device('cpu'))
 if(st.button("Click to generate image")):

 model = SimpleUnet()
 st.title("Generatig images using a diffusion model")
+model.load_state_dict(torch.load("new_linear_model_1090.pt", map_location=torch.device('cpu')))
 if(st.button("Click to generate image")):