More model options

.gitignore

@@ -1,3 +1,6 @@
+# Pycharm
+.idea
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

README.md

@@ -1,7 +1,7 @@
 ---
 title: Cite Diversely
 emoji: 🎓
-colorFrom: orange
+colorFrom: yellow
 colorTo: gray
 sdk: streamlit
 sdk_version: 1.10.0 # The latest supported version

app.py

@@ -34,7 +34,7 @@ class References(object):
     def infer_ethnicity(self):
         self.raw_results = ethnicolr.pred_census_ln(self.raw_results, 'Last Name', 2010)
         self.raw_results['Most Likely Ethnicity'] = self.raw_results['race']
-        self.raw_results.drop(labels=['race', 'white', 'black', 'hispanic', 'api'])
+        self.raw_results.drop(columns=['race', 'white', 'black', 'hispanic', 'api'])
 
         for i in self.raw_results['Most Likely Ethnicity']:
             self.ethnicity_results[i] = self.ethnicity_results.get(i, 0) + 1