Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import TextIteratorStreamer
 import spaces
 
 model_name = "numfa/numfa_v2-3b"
-model = AutoModelForCausalLM.from_pretrained(model_name,torch_dtype=torch.float16, device="cuda")
+model = AutoModelForCausalLM.from_pretrained(model_name,torch_dtype=torch.float16, device_map="cuda")
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 if tokenizer.pad_token_id is None:
     tokenizer.pad_token_id = tokenizer.eos_token_id