Update app.py

app.py

@@ -28,7 +28,7 @@ class _MLPVectorProjector(nn.Module):
 model_name = "microsoft/phi-2"
 
 with torch.no_grad():
-    phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="auto",torch.dtype=torch.float16)
+    phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="auto",torch_dtype=torch.float16)
      
 tokenizer_text = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)