Spaces:

goendalf666
/

test

Runtime error

goendalf666 commited on Oct 3, 2023

Commit

37fb80b

•

1 Parent(s): 8a396c1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 # Initialize the model and tokenizer
 cuda = "cuda:0" if torch.cuda.is_available() else "cpu"
-model = AutoModelForCausalLM.from_pretrained("goendalf666/salesGPT_v2").to(cuda)
 tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5")
 def interact_with_model(user_input):

 # Initialize the model and tokenizer
 cuda = "cuda:0" if torch.cuda.is_available() else "cpu"
+model = AutoModelForCausalLM.from_pretrained("goendalf666/salesGPT_v2", trust_remote_code=True).to(cuda)
 tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5")
 def interact_with_model(user_input):