Chris4K commited on
Commit
a9006e9
·
verified ·
1 Parent(s): 361ef82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ login(token=hf_token)
45
  # Define the model pipeline with additional generation parameters
46
  model_pipeline = pipeline(
47
  # model="meta-llama/Llama-3.2-1B",
48
- model="sentence-transformers/all-MiniLM-L6-v2"
49
  #use_auth_token=hf_token,
50
  max_length=1000, # You can increase this if needed
51
  max_new_tokens=500 # Limit how many tokens are generated
 
45
  # Define the model pipeline with additional generation parameters
46
  model_pipeline = pipeline(
47
  # model="meta-llama/Llama-3.2-1B",
48
+ model="sentence-transformers/all-MiniLM-L6-v2",
49
  #use_auth_token=hf_token,
50
  max_length=1000, # You can increase this if needed
51
  max_new_tokens=500 # Limit how many tokens are generated