Initial commit

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/vcs.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings" defaultProject="true" />
+</project>

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.11-slim
+
+WORKDIR /code
+
+RUN apt-get update && apt-get install -y procps && pip install -U pip && rm /etc/localtime && ln -s /usr/share/zoneinfo/America/Mexico_City /etc/localtime
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install -r ./requirements.txt
+
+COPY ./main.py /code/
+
+EXPOSE 8501
+
+CMD ["streamlit","run", "main.py"]

main.py

@@ -0,0 +1,34 @@
+import streamlit as st
+import requests
+import json
+
+st.write("""
+# Application to predict the time for the NYC taxi trips
+""")
+
+st.sidebar.header('User Input Parameters')
+
+def user_input_features():
+    PU = st.sidebar.text_input("PU Location ID", "80")
+    DO = st.sidebar.text_input("DO Location ID", "60")
+    trip_distance = st.sidebar.number_input("Trip Distance", value=10.0, min_value=0.1, max_value=100.0)
+
+    input_dict = {
+        'PULocationID': PU,
+        'DOLocationID': DO,
+        'trip_distance': trip_distance,
+    }
+
+    return input_dict
+
+
+input_dict = user_input_features()
+
+if st.button('Predict'):
+    response = requests.post(
+        #url="http://localhost:8000/predict",
+        url="http://nyc-taxi-model-container:8000/predict",
+        data=json.dumps(input_dict)
+    )
+
+    st.write(f"El tiempo estimado del viaje es: {response.json()['prediction']} minutos")

requirements.txt

@@ -0,0 +1,2 @@
+requests==2.32.3
+streamlit==1.39.0