etuncer commited on
Commit
9eebd37
·
verified ·
1 Parent(s): 47277ec

Rename src/streamlit_app.py to src/app.py

Browse files
Files changed (2) hide show
  1. src/app.py +27 -0
  2. src/streamlit_app.py +0 -40
src/app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ import streamlit as st
3
+ from PIL import Image, ImageOps
4
+ import numpy as np
5
+
6
+ # Load the model
7
+ model = joblib.load('src/mnist_svc_model.pkl')
8
+
9
+ # Streamlit app
10
+ st.title("🔢 MNIST Digits Classification")
11
+ st.write("Predict the digit of a handwritten digit.")
12
+ st.image("https://opendatascience.com/wp-content/uploads/2017/05/handwritten.jpg", width='stretch')
13
+
14
+ # Image input
15
+ image = st.file_uploader("Upload an image of a handwritten digit", type=["jpg", "jpeg", "png"])
16
+ if image:
17
+ st.image(image, width=200)
18
+
19
+ # Predict button
20
+ if st.button("Predict", type="primary", use_container_width=True):
21
+ if image:
22
+ img = ImageOps.invert(Image.open(image).convert('L').resize((28, 28)))
23
+ x = np.array(img, dtype=np.float32).reshape(1, -1) / 255.0
24
+ prediction = model.predict(x)
25
+ st.success(f"Predicted digit: **{prediction[0]}**")
26
+ else:
27
+ st.error("Please upload an image of a handwritten digit")
src/streamlit_app.py DELETED
@@ -1,40 +0,0 @@
1
- import altair as alt
2
- import numpy as np
3
- import pandas as pd
4
- import streamlit as st
5
-
6
- """
7
- # Welcome to Streamlit!
8
-
9
- Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
10
- If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
11
- forums](https://discuss.streamlit.io).
12
-
13
- In the meantime, below is an example of what you can do with just a few lines of code:
14
- """
15
-
16
- num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
17
- num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
18
-
19
- indices = np.linspace(0, 1, num_points)
20
- theta = 2 * np.pi * num_turns * indices
21
- radius = indices
22
-
23
- x = radius * np.cos(theta)
24
- y = radius * np.sin(theta)
25
-
26
- df = pd.DataFrame({
27
- "x": x,
28
- "y": y,
29
- "idx": indices,
30
- "rand": np.random.randn(num_points),
31
- })
32
-
33
- st.altair_chart(alt.Chart(df, height=700, width=700)
34
- .mark_point(filled=True)
35
- .encode(
36
- x=alt.X("x", axis=None),
37
- y=alt.Y("y", axis=None),
38
- color=alt.Color("idx", legend=None, scale=alt.Scale()),
39
- size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
40
- ))