Commit
•
21fe431
1
Parent(s):
dd4f18c
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from transformers import pipeline
|
|
| 9 |
pipe = pipeline("text-generation", model="KvrParaskevi/Llama-2-7b-Hotel-Booking-Model-8Bit")
|
| 10 |
|
| 11 |
def predict(text):
|
| 12 |
-
|
| 13 |
print(answer)
|
| 14 |
return pipe(text)[0]["text"]
|
| 15 |
|
|
|
|
| 9 |
pipe = pipeline("text-generation", model="KvrParaskevi/Llama-2-7b-Hotel-Booking-Model-8Bit")
|
| 10 |
|
| 11 |
def predict(text):
|
| 12 |
+
answer = pipe(text)[0]
|
| 13 |
print(answer)
|
| 14 |
return pipe(text)[0]["text"]
|
| 15 |
|