Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,6 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
|
| 3 |
|
| 4 |
import streamlit as st
|
| 5 |
|
| 6 |
-
|
| 7 |
# Load the tokenizer and model
|
| 8 |
tokenizer = GPT2Tokenizer.from_pretrained('webtoon_tokenizer')
|
| 9 |
model = GPT2LMHeadModel.from_pretrained('webtoon_model')
|
|
@@ -50,4 +49,4 @@ def generate_description(title):
|
|
| 50 |
|
| 51 |
|
| 52 |
if __name__ == '__main__':
|
| 53 |
-
main()
|
|
|
|
| 3 |
|
| 4 |
import streamlit as st
|
| 5 |
|
|
|
|
| 6 |
# Load the tokenizer and model
|
| 7 |
tokenizer = GPT2Tokenizer.from_pretrained('webtoon_tokenizer')
|
| 8 |
model = GPT2LMHeadModel.from_pretrained('webtoon_model')
|
|
|
|
| 49 |
|
| 50 |
|
| 51 |
if __name__ == '__main__':
|
| 52 |
+
main()
|