Update app.py

app.py

@@ -15,12 +15,62 @@ model_dir = model.download()
 model = joblib.load(model_dir + "/incident_model.pkl")
 
 
-def incident(pclass, sex, age, sibsp, parch, fare, embarked):
+def incident(dayofweek, reportcode, district, latitude, longitude, month, year, hour):
     input_list = []
-    if sex == 'female':
+    if dayofweek == 'Friday':
         input_list.append(1.0)
         input_list.append(0.0)
-    elif sex == 'male':
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+    elif dayofweek == 'Monday':
+        input_list.append(0.0)
+        input_list.append(1.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+    elif dayofweek == 'Saturday':
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(1.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+    elif dayofweek == 'Sunday':
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(1.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+    elif dayofweek == 'Thursday':
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(1.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+    elif dayofweek == 'Tuesday':
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(1.0)
+        input_list.append(0.0)
+    elif dayofweek == 'Wednesday':
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
+        input_list.append(0.0)
         input_list.append(0.0)
         input_list.append(1.0)
     else:
@@ -70,8 +120,8 @@ demo = gr.Interface(
         gr.inputs.Number(default="Northern", label="Police District (Northern, Bayview, Southern, Mission, Ingleside, Tenderloin, Taraval, Central, Richmond, Park)"),
         gr.inputs.Number(default=1.0, label="latitude"),
         gr.inputs.Number(default=1.0, label="longitude"),
-        gr.inputs.Number(default=2023, label="Incident Year (e.g 2019)"),
         gr.inputs.Number(default=1, label="Incident Month (1-12)"),
+        gr.inputs.Number(default=2023, label="Incident Year (e.g 2019)"),
         gr.inputs.Number(default=1, label="Incident Hour (0-23)"),
     ],
     outputs=gr.Image(type="pil"))