update space

app.py

@@ -265,7 +265,7 @@ def main():
             e_type = 'cyber'
 
         model = models[e_type]
-        model = model.to(device)
+        print("model", model)
         # model = load_model('text300M', device=device)
         # with torch.no_grad():
         #     new_proj = nn.Linear(1024 * 2, 1024, device=device, dtype=model.wrapped.input_proj.weight.dtype)