abhisheky127 commited on
Commit
ae9e0c1
1 Parent(s): 947bf26

updating preprocess function

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -12,6 +12,8 @@ classifier = pipeline("zero-shot-classification",
12
  def zero_shot(doc, candidates):
13
  given_labels = candidates.split(",")
14
  given_labels = list(map(str.strip, given_labels))
 
 
15
  dictionary = classifier(preprocess(doc), given_labels)
16
  labels = dictionary['labels']
17
  scores = dictionary['scores']
@@ -38,9 +40,9 @@ output = gr.Label(label="Output")
38
 
39
  #example object
40
  transactions_and_tags = [
41
- ["MPS/TRUFFLES /202303261700/034587/Bangalore", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
42
- ["MPS/TACO BELL /202304012247/108300/BANGALORE", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
43
- ["POS XXXXXXXXXXXX0001 APOLLO PHARMACY", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
44
  ["BIL/ONL/000471093694/1MG Techno/X7ZRUSVLURFQZO", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
45
  ["POS XXXXXXXXXXXX1111 DECATHLON SPORTS", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
46
  ["POS XXXXXXXXXXXX1111 IKEA INDIA PVT L", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
 
12
  def zero_shot(doc, candidates):
13
  given_labels = candidates.split(",")
14
  given_labels = list(map(str.strip, given_labels))
15
+ doc = preprocess(doc)
16
+ print(doc)
17
  dictionary = classifier(preprocess(doc), given_labels)
18
  labels = dictionary['labels']
19
  scores = dictionary['scores']
 
40
 
41
  #example object
42
  transactions_and_tags = [
43
+ ["MPS/TRUFFLES/202303261700/034587/Bangalore", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
44
+ ["MPS/TACO BELL/202304012247/108300/BANGALORE", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
45
+ ["POS XXXXXXXXXXXX0001APOLLO PHARMACY", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
46
  ["BIL/ONL/000471093694/1MG Techno/X7ZRUSVLURFQZO", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
47
  ["POS XXXXXXXXXXXX1111 DECATHLON SPORTS", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],
48
  ["POS XXXXXXXXXXXX1111 IKEA INDIA PVT L", "Medical, Food, Shopping, Subscription, Travel, Miscellaneous"],