matheuscvp commited on
Commit
b4513b7
·
1 Parent(s): 59ec0fd
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def predict(wind, temp_max, temp_min, precipitation):
42
  prediction = model.predict(df)
43
  prediction2 = model2.predict(df)
44
 
45
- return [options[round(prediction)], options[round(prediction2)]]
46
 
47
  iface = gr.Interface(
48
  fn=predict,
 
42
  prediction = model.predict(df)
43
  prediction2 = model2.predict(df)
44
 
45
+ return [options[round(max(prediction))], options[round(max(prediction2))]]
46
 
47
  iface = gr.Interface(
48
  fn=predict,