Aliou12 commited on
Commit
919bf04
·
1 Parent(s): 7a7942a

new application

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -31,7 +31,7 @@ def predict_price(kms_driven, present_price, fuel_type, seller_type, transmissio
31
 
32
  # Prédiction
33
  prediction = lr.predict(features)[0]
34
- return round(prediction, 2)
35
  except Exception as e:
36
  return f"Erreur lors de la prédiction : {str(e)}"
37
 
@@ -65,10 +65,10 @@ def predict_from_csv(file):
65
  features = data[required_columns].values
66
  predictions = lr.predict(features)
67
 
68
- # Calculer la moyenne des prédictions
69
- average_prediction = np.mean(predictions)
70
 
71
- return round(average_prediction, 2)
72
  except Exception as e:
73
  return f"Erreur lors de la prédiction : {str(e)}"
74
 
@@ -85,7 +85,7 @@ simple_output = gr.Number(label="Predicted Price")
85
 
86
  # Interface Gradio pour la prédiction multiple via un fichier CSV
87
  csv_input = gr.File(label="Déposez votre fichier CSV")
88
- csv_output = gr.Number(label="Moyenne des prix prévus")
89
 
90
  # Interface combinée
91
  interface = gr.TabbedInterface(
@@ -102,7 +102,7 @@ interface = gr.TabbedInterface(
102
  inputs=csv_input,
103
  outputs=csv_output,
104
  title="Car Price Prediction (Multiple)",
105
- description="Téléversez un fichier CSV contenant les données des voitures pour obtenir la moyenne des prédictions de prix.",
106
  ),
107
  ],
108
  tab_names=["Prédiction Unique", "Prédiction Multiple"]
 
31
 
32
  # Prédiction
33
  prediction = lr.predict(features)[0]
34
+ return max(round(prediction, 2), 0) # S'assurer que le prix prédit n'est pas négatif
35
  except Exception as e:
36
  return f"Erreur lors de la prédiction : {str(e)}"
37
 
 
65
  features = data[required_columns].values
66
  predictions = lr.predict(features)
67
 
68
+ # Ajouter les prédictions au DataFrame
69
+ data["Predicted_Price"] = [max(pred, 0) for pred in predictions] # S'assurer que les prix prédits ne sont pas négatifs
70
 
71
+ return data
72
  except Exception as e:
73
  return f"Erreur lors de la prédiction : {str(e)}"
74
 
 
85
 
86
  # Interface Gradio pour la prédiction multiple via un fichier CSV
87
  csv_input = gr.File(label="Déposez votre fichier CSV")
88
+ csv_output = gr.Dataframe(label="Prédictions avec fichier CSV")
89
 
90
  # Interface combinée
91
  interface = gr.TabbedInterface(
 
102
  inputs=csv_input,
103
  outputs=csv_output,
104
  title="Car Price Prediction (Multiple)",
105
+ description="Téléversez un fichier CSV contenant les données des voitures pour obtenir les prédictions de prix.",
106
  ),
107
  ],
108
  tab_names=["Prédiction Unique", "Prédiction Multiple"]