Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,24 +10,7 @@ from transformers import pipeline
|
|
| 10 |
summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
|
| 11 |
|
| 12 |
# Example input for inference
|
| 13 |
-
dialogue=
|
| 14 |
-
'''
|
| 15 |
-
Hannah: Hey, do you have Betty's number?
|
| 16 |
-
Amanda: Lemme check
|
| 17 |
-
Hannah: <file_gif>
|
| 18 |
-
Amanda: Sorry, can't find it.
|
| 19 |
-
Amanda: Ask Larry
|
| 20 |
-
Amanda: He called her last time we were at the park together
|
| 21 |
-
Hannah: I don't know him well
|
| 22 |
-
Hannah: <file_gif>
|
| 23 |
-
Amanda: Don't be shy, he's very nice
|
| 24 |
-
Hannah: If you say so..
|
| 25 |
-
Hannah: I'd rather you texted him
|
| 26 |
-
Amanda: Just text him 🙂
|
| 27 |
-
Hannah: Urgh.. Alright
|
| 28 |
-
Hannah: Bye
|
| 29 |
-
Amanda: Bye bye
|
| 30 |
-
'''
|
| 31 |
# Perform inference
|
| 32 |
summary = summarizer(dialogue, max_length=500, min_length=300, do_sample=False)
|
| 33 |
|
|
|
|
| 10 |
summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
|
| 11 |
|
| 12 |
# Example input for inference
|
| 13 |
+
dialogue = input(str("Enter the input:"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
# Perform inference
|
| 15 |
summary = summarizer(dialogue, max_length=500, min_length=300, do_sample=False)
|
| 16 |
|