rahimizadeh commited on
Commit
12ff613
·
verified ·
1 Parent(s): 9be32e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -18,14 +18,14 @@ class TextSummarizer:
18
  model_id=self.model_id,
19
  task="summarization",
20
  pipeline_kwargs={
21
- "max_length": 250,
22
- "do_sample": True,
23
- "temperature": 0.7, # More creative
24
- "top_k": 50, # Limit to top 50 tokens
25
- "top_p": 0.95 # Use nucleus sampling
26
- #"max_length": max_length,
27
- #"min_length": min_length,
28
- #"do_sample": False # Deterministic output
29
  }
30
  )
31
 
 
18
  model_id=self.model_id,
19
  task="summarization",
20
  pipeline_kwargs={
21
+ #"max_length": 250,
22
+ #"do_sample": True,
23
+ #"temperature": 0.7, # More creative
24
+ #"top_k": 50, # Limit to top 50 tokens
25
+ #"top_p": 0.95 # Use nucleus sampling
26
+ "max_length": max_length,
27
+ "min_length": min_length,
28
+ "do_sample": False # Deterministic output
29
  }
30
  )
31