Spaces:
Paused
Paused
prompt tweak
Browse files
app.py
CHANGED
@@ -17,7 +17,6 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
17 |
model.to(device)
|
18 |
|
19 |
|
20 |
-
|
21 |
other_benifits = '''Extract the following information in the given format:
|
22 |
{'other_benefits_and_information': {
|
23 |
'401k eru: {'This Period':'', 'Year-to-Date':''}},
|
@@ -26,8 +25,8 @@ other_benifits = '''Extract the following information in the given format:
|
|
26 |
'sick:': '',
|
27 |
'vacation:': '',
|
28 |
}
|
29 |
-
'payment method': '',
|
30 |
-
'Amount': ''
|
31 |
}
|
32 |
'''
|
33 |
|
@@ -39,7 +38,7 @@ tax_deductions = '''Extract the following information in the given format:
|
|
39 |
'ee social security tax:': {'Amount':'', 'Year-To_Date':""},
|
40 |
'ee medicare tax:': {'Amount':'', 'Year-To_Date':""}},
|
41 |
'california:': {
|
42 |
-
'withholding tax:': {'Amount':'', 'Year-To_Date':""},
|
43 |
'ee disability tax:': {'Amount':'', 'Year-To_Date':""}}},
|
44 |
}
|
45 |
'''
|
|
|
17 |
model.to(device)
|
18 |
|
19 |
|
|
|
20 |
other_benifits = '''Extract the following information in the given format:
|
21 |
{'other_benefits_and_information': {
|
22 |
'401k eru: {'This Period':'', 'Year-to-Date':''}},
|
|
|
25 |
'sick:': '',
|
26 |
'vacation:': '',
|
27 |
}
|
28 |
+
'payment method': 'eg. Direct payment',
|
29 |
+
'Amount': 'eg. 12.99'
|
30 |
}
|
31 |
'''
|
32 |
|
|
|
38 |
'ee social security tax:': {'Amount':'', 'Year-To_Date':""},
|
39 |
'ee medicare tax:': {'Amount':'', 'Year-To_Date':""}},
|
40 |
'california:': {
|
41 |
+
'withholding tax:': {'Amount':'', 'Year-To_Date':""},
|
42 |
'ee disability tax:': {'Amount':'', 'Year-To_Date':""}}},
|
43 |
}
|
44 |
'''
|