Update app.py

app.py

@@ -36,7 +36,8 @@ class LayerWrapper(torch.nn.Module):
         return layer_outputs
 
 moondream.text_model.transformer.h = torch.nn.ModuleList([
-    LayerWrapper(layer, vector, 4.2) for layer, vector in zip(og_h, control_vectors)
+    LayerWrapper(layer, vector, 4.2)
+    for layer, vector in zip(moondream.text_model.transformer.h, control_vectors)
 ])
 
 @spaces.GPU(duration=10)