Update app.py

app.py

@@ -12,7 +12,7 @@ import os
 from transformers import pipeline
 import itertools
 import pandas as pd
-import thefuzz
+
 
 OUT_HEADERS = ['E','S','G']
 DF_SP500 = pd.read_csv('SP500_constituents.zip',compression=dict(method='zip'))
@@ -29,12 +29,13 @@ MODEL_SENTIMENT_ANALYSIS = "ProsusAI/finbert"
 def get_company_sectors(extracted_names, threshold=0.95):
   '''
   '''
+  from thefuzz import process, fuzz
   output = []
   standard_names_tuples = []
   for extracted_name in extracted_names:
-    name_match = thefuzz.process.extractOne(extracted_name,
+    name_match = process.extractOne(extracted_name,
                                             DF_SP500.Name, 
-                                            scorer=thefuzz.fuzz.token_set_ratio)
+                                            scorer=fuzz.token_set_ratio)
     similarity = name_match[1]/100
     if similarity >= threshold:
       standard_names_tuples.append(name_match[:2])