Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ if user_input:
|
|
52 |
# Get response from model with correct explicit parameters
|
53 |
llm = HuggingFaceEndpoint(
|
54 |
repo_id="google/pegasus-xsum",
|
55 |
-
task="
|
56 |
temperature=0.3, # Explicitly passing temperature here
|
57 |
max_new_tokens=128 # Explicitly passing max_new_tokens here
|
58 |
)
|
|
|
52 |
# Get response from model with correct explicit parameters
|
53 |
llm = HuggingFaceEndpoint(
|
54 |
repo_id="google/pegasus-xsum",
|
55 |
+
task="summarization", # Change to 'summarization'
|
56 |
temperature=0.3, # Explicitly passing temperature here
|
57 |
max_new_tokens=128 # Explicitly passing max_new_tokens here
|
58 |
)
|