Hugging Face's logo

Spaces:
LunaticMaestro
/
book-recommender
Sleeping

App Files Community
Deepak Sahu commited on
Commit
86af19f
·
1 Parent(s): bafb8fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,14 +8,14 @@
8
  # def get_recommendation(book_title: str) -> str:
9
  # return book_title
10
 
11
- from transformers import pipeline, set_seed
12
 
13
- # CONST
14
- set_seed(42)
15
- TRAINED_CASUAL_MODEL = "LunaticMaestro/gpt2-book-summary-generator"
16
 
17
 
18
- generator_model = pipeline('text-generation', model=TRAINED_CASUAL_MODEL)
19
 
20
 
21
 
 
8
  # def get_recommendation(book_title: str) -> str:
9
  # return book_title
10
 
11
+ # from transformers import pipeline, set_seed
12
 
13
+ # # CONST
14
+ # set_seed(42)
15
+ # TRAINED_CASUAL_MODEL = "LunaticMaestro/gpt2-book-summary-generator"
16
 
17
 
18
+ # generator_model = pipeline('text-generation', model=TRAINED_CASUAL_MODEL)
19
 
20
 
21