Spaces:
Runtime error
Runtime error
Commit
•
b358447
1
Parent(s):
4ea6b57
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,10 @@ pipeline = pipeline("feature-extraction", model="WhereIsAI/UAE-Large-V1")
|
|
| 5 |
|
| 6 |
|
| 7 |
def predict(text):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
title_outputs = pipeline(text)
|
| 9 |
title_outputs = torch.tensor(title_outputs)
|
| 10 |
|
|
|
|
| 5 |
|
| 6 |
|
| 7 |
def predict(text):
|
| 8 |
+
text = text.lower()
|
| 9 |
+
text = text.translate(str.maketrans('', '', string.punctuation))
|
| 10 |
+
|
| 11 |
+
|
| 12 |
title_outputs = pipeline(text)
|
| 13 |
title_outputs = torch.tensor(title_outputs)
|
| 14 |
|