Spaces:
Sleeping
Sleeping
Deepak Sahu
commited on
Commit
·
59b441e
1
Parent(s):
2c620b0
Update z_hypothetical_summary.py
Browse files
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 |
-
|
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]:
|