Chris4K commited on
Commit
361ef82
1 Parent(s): 21db1c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -44,7 +44,8 @@ login(token=hf_token)
44
 
45
  # Define the model pipeline with additional generation parameters
46
  model_pipeline = pipeline(
47
- model="meta-llama/Llama-3.2-1B",
 
48
  #use_auth_token=hf_token,
49
  max_length=1000, # You can increase this if needed
50
  max_new_tokens=500 # Limit how many tokens are generated
 
44
 
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