Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,8 @@ pipeline = pipeline(model="jeevana/EmailSubjectLineGeneration", max_new_tokens=1
|
|
| 8 |
def predict(input):
|
| 9 |
print("pipeline object", pipeline)
|
| 10 |
prediction = pipeline(input)
|
| 11 |
-
|
|
|
|
| 12 |
return prediction
|
| 13 |
|
| 14 |
|
|
|
|
| 8 |
def predict(input):
|
| 9 |
print("pipeline object", pipeline)
|
| 10 |
prediction = pipeline(input)
|
| 11 |
+
prediction = prediction[0].get("generated_text")
|
| 12 |
+
prediction = prediction[len(input)+8]
|
| 13 |
return prediction
|
| 14 |
|
| 15 |
|