Update app.py

app.py

@@ -32,6 +32,11 @@ def classify_class(incident_description):
     classification = response.choices[0].text.strip()
     return classification
 
+def main(incident_description):
+    defect_class = classify_class(incident_description)
+    main_issue =  classify_cause(incident_description)
+    return defect_class, main_issue
+    
 inputs =  gr.inputs.Textbox(label="Flight Incident Description")
 outputs = [gr.outputs.Textbox(label="Main Issue of the flight incident"),
            gr.outputs.Textbox(label="category of the flight incident")]