Upload 4 files

app.py

@@ -0,0 +1,45 @@
+import streamlit as st
+import pandas as pd
+import joblib
+from sklearn.linear_model import LinearRegression
+from sklearn.tree import DecisionTreeRegressor
+from sklearn.ensemble import RandomForestRegressor
+
+# Load the trained models (make sure these pickle files are uploaded in the same directory as your app.py in Hugging Face Space)
+lr = joblib.load('linear_regression_model.pkl')
+dt = joblib.load('decision_tree_model.pkl')
+rf = joblib.load('random_forest_model.pkl')
+
+# Streamlit UI
+st.title("Indian Food Cook Time Prediction")
+st.write("Enter the features to predict the cook time of Indian food.")
+
+# Input fields
+diet = st.selectbox("Diet Type", options=["vegetarian", "non vegetarian"])
+prep_time = st.number_input("Preparation Time (minutes)", min_value=0, step=1)
+ingredients = st.number_input("Number of Ingredients", min_value=1, step=1)
+
+# Convert diet to numeric (0 for vegetarian, 1 for non-vegetarian)
+diet = 0 if diet == "vegetarian" else 1
+
+# Combine inputs into a DataFrame for prediction
+input_data = pd.DataFrame({
+    'diet': [diet],
+    'prep_time': [prep_time],
+    'num_ingredients': [ingredients]
+})
+
+# Model selection
+model_choice = st.selectbox("Select a Model", ["Linear Regression", "Decision Tree", "Random Forest"])
+
+if st.button("Predict Cook Time"):
+    # Make prediction based on model choice
+    if model_choice == "Linear Regression":
+        prediction = lr.predict(input_data)[0]
+    elif model_choice == "Decision Tree":
+        prediction = dt.predict(input_data)[0]
+    elif model_choice == "Random Forest":
+        prediction = rf.predict(input_data)[0]
+
+    # Display the predicted cook time
+    st.write(f"Predicted Cook Time: {round(prediction, 2)} minutes")

decision_tree_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dad5e8aa2bd01508be87dcbfa08a67e3c86a0a7589b51b22d6709b04f9133f09
+size 8929

linear_regression_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9badbd8144112a7358f14a306a88df1bbe24efa6743a57c75b4687ff807ff259
+size 928

random_forest_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb8f1b31d3a1032369038017849b85de8839188af50eb2d6369333b6104a0d43
+size 611169