Add specific examples
Browse files- inference.py +7 -1
inference.py
CHANGED
|
@@ -16,4 +16,10 @@ def predict(question, context):
|
|
| 16 |
|
| 17 |
answer['answer'] = text
|
| 18 |
|
| 19 |
-
return answer
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
answer['answer'] = text
|
| 18 |
|
| 19 |
+
return answer
|
| 20 |
+
|
| 21 |
+
if __name__ == '__main__':
|
| 22 |
+
question = 'Combi cao bao nhiêu.'
|
| 23 |
+
context = 'Combi là sinh viên năm 2 trường Ecole Polytechnique. Chiều cao của Combi là 1m73, cân nặng là 63kg. Combi thích học Machine Learning vì Machine Learning cần nhiều toán.'
|
| 24 |
+
|
| 25 |
+
print(predict(question, context))
|