fixed radio buttons

app.py

@@ -23,11 +23,51 @@ print("Model downloaded")
 
 
 def predict(magnitudeOfDelay, hour, iconCategory, latitude, longitude, month):
+    # Change the magnitude delay to an integer based on the index
+    if magnitudeOfDelay == 'Unknown':
+        magnitudeOfDelay = 0
+    elif magnitudeOfDelay == 'Minor':
+        magnitudeOfDelay = 1
+    elif magnitudeOfDelay == 'Moderate':
+        magnitudeOfDelay = 2
+    elif magnitudeOfDelay == 'Major':
+        magnitudeOfDelay = 3
+    elif magnitudeOfDelay == 'Undefined':
+        magnitudeOfDelay = 4
+
+    # Change the icon category to an integer based on the index
+    if iconCategory == 'Unknown':
+        iconCategory = 0
+    elif iconCategory == 'Accident':
+        iconCategory = 1
+    elif iconCategory == 'Fog':
+        iconCategory = 2
+    elif iconCategory == 'Dangerous Conditions':
+        iconCategory = 3
+    elif iconCategory == 'Rain':
+        iconCategory = 4
+    elif iconCategory == 'Ice':
+        iconCategory = 5
+    elif iconCategory == 'Jam':
+        iconCategory = 6
+    elif iconCategory == 'Lane Closed':
+        iconCategory = 7
+    elif iconCategory == 'Road Closed':
+        iconCategory = 8
+    elif iconCategory == 'Road Works':
+        iconCategory = 9
+    elif iconCategory == 'Wind':
+        iconCategory = 10
+    elif iconCategory == 'Flooding':
+        iconCategory = 11
+    elif iconCategory == 'Broken Down Vehicle':
+        iconCategory = 14
+
     # Create a row from the input
     row = {
-        'magnitudeOfDelay': magnitudeOfDelay,
-        'hour': hour,
-        'iconCategory': iconCategory,
+        'magnitudeOfDelay': int(magnitudeOfDelay),
+        'hour': int(hour),
+        'iconCategory': int(iconCategory),
         'latitude': latitude,
         'longitude': longitude,
         'month': month
@@ -57,9 +97,9 @@ demo = gr.Interface(
     description="Predicts the duration of a traffic incident in Stockholm",
     allow_flagging="never",
     inputs=[
-        gr.inputs.Slider(0, 60, label="Magnitude of Delay"),
+        gr.inputs.Radio(["Unknown", "Minor", "Moderate", "Major", "Undefined"]),
         gr.inputs.Slider(0, 23, label="Hour"),
-        gr.inputs.Radio(["Accident", "Construction", "Congestion", "Disabled Vehicle", "Mass Transit", "Miscellaneous", "Other News", "Planned Event", "Road Hazard", "Roadwork", "Traffic Flow", "Weather"], label="Icon Category"),
+        gr.inputs.Radio(["Unknown", "Accident", "Fog", "Dangerous Conditions", "Rain", "Ice", "Jam", "Lane Closed", "Road Closed", "Road Works", "Wind", "Flooding", "Broken Down Vehicle"], label="Icon Category"),
         gr.inputs.Slider(59.25, 59.40, label="Latitude"),
         gr.inputs.Slider(18.00, 18.16, label="Longitude"),
         gr.inputs.Slider(1, 12, label="Month")