SaniaE commited on
Commit
6b4af39
·
verified ·
1 Parent(s): 7397034

added debugging steps

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -89,9 +89,10 @@ def generate_random():
89
 
90
  with torch.inference_mode():
91
  torch.seed()
92
-
93
  noise = torch.randn(1, Z_DIM, device=DEVICE)
 
94
  fake_img = gen_model(noise)
 
95
  return StreamingResponse(get_image_stream(fake_img), media_type="image/png")
96
 
97
 
@@ -115,6 +116,7 @@ def explore_latent(seed: int, x_shift: float = Query(0.0, ge=-5.0, le=5.0), y_sh
115
  direction = torch.randn_like(noise)
116
  noise = noise + 0.3 * direction * (abs(x_shift) + abs(y_shift))
117
 
 
118
  fake_img = gen_model(noise)
119
  return StreamingResponse(get_image_stream(fake_img), media_type="image/png")
120
 
 
89
 
90
  with torch.inference_mode():
91
  torch.seed()
 
92
  noise = torch.randn(1, Z_DIM, device=DEVICE)
93
+ print("NOISE:", noise[0, :5])
94
  fake_img = gen_model(noise)
95
+ print("SEED: ", seed)
96
  return StreamingResponse(get_image_stream(fake_img), media_type="image/png")
97
 
98
 
 
116
  direction = torch.randn_like(noise)
117
  noise = noise + 0.3 * direction * (abs(x_shift) + abs(y_shift))
118
 
119
+ print("NOISE:", noise[0, :5])
120
  fake_img = gen_model(noise)
121
  return StreamingResponse(get_image_stream(fake_img), media_type="image/png")
122