Update app.py

app.py

@@ -72,7 +72,11 @@ def humands(Sex,Age,Married,Monthlyincome,TotalWorkingYears,DistanceFromHome,Ove
     #    predicted="La nota media que se puede lograr en el grupo serà de "+ str(round(pred,2)) + " . Vais por buen camino"
     #else:
     #    predicted="La nota media que se puede lograr en el grupo serà de "+ str(round(pred,2)) + " . Buen trabajo"
-    return pred
+    if pred == "Yes":
+        predicted="stressed_image.jpg"
+    else:
+        predicted="ok_image.jpg"
+    return predicted
    # return predicted
     
 iface = gr.Interface(
@@ -90,7 +94,7 @@ iface = gr.Interface(
         
      ],
 
-    "text",
+    [gr.outputs.Image(type='file')],
     examples=[
         ["Mujer",25,"Soltero",1500,20,2,True,2,2],
         ["Hombre",25,"Casado",1500,20,2,False,2,2],