Trying again

app.py

@@ -4,9 +4,10 @@ from transformers import pipeline
 # Load the pipeline
 tourModel = pipeline(model="manhan/GPT-Tour")
 
-def getTour(hh_income,hh_size,age,sex,edu,wrk): 
+def getTour(income,size,years,sex,edu,wrk): 
     # person = a dict with person-level and hh-level attributes:
     person = {}
+    hh_income = int(income)
     if hh_income<0:
         _skip = 1 # nvm, bad data
     elif hh_income<4: # $25,000
@@ -19,6 +20,7 @@ def getTour(hh_income,hh_size,age,sex,edu,wrk):
         person['hh_inc'] = 'high'
     else: # over
         person['hh_inc'] = 'affluent'
+    hh_size = int(size)
     if hh_size == 1:
         person['hh_size'] = 'single'
     elif hh_size == 2:
@@ -27,6 +29,7 @@ def getTour(hh_income,hh_size,age,sex,edu,wrk):
         person['hh_size'] = 'small'
     else: # more than four people
         person['hh_size'] = 'large'
+    age = int(years)
     if age < 18:
         person['age_grp'] = 'child'
     elif age < 45: