henry2024 commited on
Commit
d32a246
·
verified ·
1 Parent(s): 1c5e4d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
93
  time.sleep(2)
94
  return "", chat_history
95
  '''
96
- def respond(message, chat_history, base_model = "/home/henry/Desktop/ARIN7102/phi-2", device=device): # "meta-llama/Meta-Llama-3-70B"
97
  # define the model and tokenizer.
98
  # model = PhiForCausalLM.from_pretrained(base_model)
99
  model = AutoModelForCausalLM.from_pretrained(base_model)
 
93
  time.sleep(2)
94
  return "", chat_history
95
  '''
96
+ def respond(message, chat_history, base_model = "microsoft/phi-2", device=device): # "meta-llama/Meta-Llama-3-70B"
97
  # define the model and tokenizer.
98
  # model = PhiForCausalLM.from_pretrained(base_model)
99
  model = AutoModelForCausalLM.from_pretrained(base_model)