xelpmocAI commited on
Commit
ec4bd6c
·
verified ·
1 Parent(s): f742a64

prompt tweak

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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
  '''