Update app.py
Browse files
app.py
CHANGED
@@ -47,8 +47,8 @@ login(token=hf_token)
|
|
47 |
# Define the model pipeline with additional generation parameters
|
48 |
model_pipeline = pipeline(
|
49 |
# model="meta-llama/Llama-3.2-1B",
|
50 |
-
model="
|
51 |
-
pad_token_id=50256,
|
52 |
#use_auth_token=hf_token,
|
53 |
#max_length=1000, # You can increase this if needed
|
54 |
max_new_tokens=900 # Limit how many tokens are generated
|
|
|
47 |
# Define the model pipeline with additional generation parameters
|
48 |
model_pipeline = pipeline(
|
49 |
# model="meta-llama/Llama-3.2-1B",
|
50 |
+
model="dunzhang/stella_en_1.5B_v5",
|
51 |
+
#pad_token_id=50256,
|
52 |
#use_auth_token=hf_token,
|
53 |
#max_length=1000, # You can increase this if needed
|
54 |
max_new_tokens=900 # Limit how many tokens are generated
|