update space

app.py

@@ -257,7 +257,7 @@ def main():
         elif 'cyber' in instruction:
             e_type = 'cyber'
 
-        model = models[e_type]
+        model = models[e_type].to(device)
         # 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)