Deepak Sahu commited on
Commit
59b441e
·
1 Parent(s): 2c620b0

Update z_hypothetical_summary.py

Browse files
Files changed (1) hide show
  1. z_hypothetical_summary.py +1 -2
z_hypothetical_summary.py CHANGED
@@ -15,8 +15,7 @@ def load_model():
15
  global generator_model
16
  if generator_model is None:
17
  generator_model = pipeline('text-generation', model=TRAINED_CASUAL_MODEL)
18
- else:
19
- return generator_model
20
 
21
 
22
  def generate_summaries(book_title: str, genre: Optional[str] = None, n_samples=2, top_k = 50, top_p = 0.85, model=None) -> list[str]:
 
15
  global generator_model
16
  if generator_model is None:
17
  generator_model = pipeline('text-generation', model=TRAINED_CASUAL_MODEL)
18
+ return generator_model
 
19
 
20
 
21
  def generate_summaries(book_title: str, genre: Optional[str] = None, n_samples=2, top_k = 50, top_p = 0.85, model=None) -> list[str]: