jophex commited on
Commit
8651d8b
·
verified ·
1 Parent(s): c1658ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import torch
7
 
8
  # Load the model and tokenizer
9
- tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3.5-mini-instruct", trust_remote_code=True, use_auth_token=True)
10
- model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3.5-mini-instruct", trust_remote_code=True, use_auth_token=True)
11
 
12
  # Define the role prompt for advertisement assistance
13
  role_prompt = "You are an advertisement assistant. Respond professionally and helpfully to advertising-related questions.\n\n"
 
6
  import torch
7
 
8
  # Load the model and tokenizer
9
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3.5-mini-instruct", trust_remote_code=True)
10
+ model = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3.5-mini-instruct", trust_remote_code=True)
11
 
12
  # Define the role prompt for advertisement assistance
13
  role_prompt = "You are an advertisement assistant. Respond professionally and helpfully to advertising-related questions.\n\n"