henry2024 commited on
Commit
940f7b4
·
verified ·
1 Parent(s): 5f9af9c

Update app.py

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