Update app.py

app.py

@@ -73,10 +73,12 @@ def humands(Sex,Age,Married,Monthlyincome,TotalWorkingYears,DistanceFromHome,Ove
     #else:
     #    predicted="La nota media que se puede lograr en el grupo serà de "+ str(round(pred,2)) + " . Buen trabajo"
     if pred == "Yes":
-        predicted="stressed_image.jpg"
+        predicted1="Estamos ante un trabajador con alto nivel de estres."
+        predicted2="stressed_image.jpg"
     else:
-        predicted="ok_image.jpg"
-    return predicted
+        predicted1="Estamos ante un trabajador tranquilo en el trabajo."
+        predicted2="ok_image2.jpg"
+    return [predicted1,predicted2]
    # return predicted
     
 iface = gr.Interface(
@@ -94,7 +96,7 @@ iface = gr.Interface(
         
      ],
 
-    [gr.outputs.Image(type='file',shape=(120,120))],
+    ["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],