Update app.py

app.py

@@ -1,16 +1,7 @@
-import subprocess
-import sys
 import gradio as gr
 import pandas as pd
 import pickle
 
-# Check if scikit-learn is installed, and if not, install it
-try:
-    import sklearn
-except ModuleNotFoundError:
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "scikit-learn"])
-    import sklearn
-
 # Load the pre-trained model
 with open('best_model.pkl', 'rb') as model_file:
     model = pickle.load(model_file)
@@ -52,22 +43,22 @@ def predict_coffee_type(time_of_day, coffee_strength, sweetness_level, milk_type
 
     return coffee_type
 
-# Gradio Interface
+# Gradio Interface using components
 interface = gr.Interface(
     fn=predict_coffee_type,
     inputs=[
-        gr.inputs.Dropdown(['morning', 'afternoon', 'evening'], label="Time of Day"),
-        gr.inputs.Dropdown(['mild', 'regular', 'strong'], label="Coffee Strength"),
-        gr.inputs.Dropdown(['unsweetened', 'lightly sweetened', 'sweet'], label="Sweetness Level"),
-        gr.inputs.Dropdown(['none', 'regular', 'skim', 'almond'], label="Milk Type"),
-        gr.inputs.Dropdown(['hot', 'iced', 'cold brew'], label="Coffee Temperature"),
-        gr.inputs.Dropdown(['yes', 'no'], label="Flavored Coffee"),
-        gr.inputs.Dropdown(['low', 'medium', 'high'], label="Caffeine Tolerance"),
-        gr.inputs.Dropdown(['Arabica', 'Robusta', 'blend'], label="Coffee Bean"),
-        gr.inputs.Dropdown(['small', 'medium', 'large'], label="Coffee Size"),
-        gr.inputs.Dropdown(['none', 'vegan', 'lactose-intolerant'], label="Dietary Preferences")
+        gr.Dropdown(['morning', 'afternoon', 'evening'], label="Time of Day"),
+        gr.Dropdown(['mild', 'regular', 'strong'], label="Coffee Strength"),
+        gr.Dropdown(['unsweetened', 'lightly sweetened', 'sweet'], label="Sweetness Level"),
+        gr.Dropdown(['none', 'regular', 'skim', 'almond'], label="Milk Type"),
+        gr.Dropdown(['hot', 'iced', 'cold brew'], label="Coffee Temperature"),
+        gr.Dropdown(['yes', 'no'], label="Flavored Coffee"),
+        gr.Dropdown(['low', 'medium', 'high'], label="Caffeine Tolerance"),
+        gr.Dropdown(['Arabica', 'Robusta', 'blend'], label="Coffee Bean"),
+        gr.Dropdown(['small', 'medium', 'large'], label="Coffee Size"),
+        gr.Dropdown(['none', 'vegan', 'lactose-intolerant'], label="Dietary Preferences")
     ],
-    outputs=gr.outputs.Textbox(label="Recommended Coffee Type"),
+    outputs=gr.Textbox(label="Recommended Coffee Type"),
     title="Coffee Type Recommendation"
 )