ZennethKenneth commited on
Commit
e703334
·
verified ·
1 Parent(s): dae9913
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ For more information on `huggingface_hub` Inference API support, please check th
15
  # requires space hardware update to use large models (TODO)
16
  # client = InferenceClient("mistralai/Mistral-Large-Instruct-2407")
17
  # Note change in instantiation***
18
- text_generator = pipeline("text-generation", model="google/gemma-2-2b", use_auth_token=hf_token)
19
 
20
  def authenticate_and_generate(token, message, history, system_message, max_tokens, temperature, top_p):
21
  # Initialize the text-generation pipeline with the provided token
 
15
  # requires space hardware update to use large models (TODO)
16
  # client = InferenceClient("mistralai/Mistral-Large-Instruct-2407")
17
  # Note change in instantiation***
18
+ text_generator = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", use_auth_token=hf_token)
19
 
20
  def authenticate_and_generate(token, message, history, system_message, max_tokens, temperature, top_p):
21
  # Initialize the text-generation pipeline with the provided token