uplaod 2 files

app.py

@@ -0,0 +1,38 @@
+import streamlit as st
+from PIL import Image
+import numpy as np
+import tensorflow as tf
+
+# 事前訓練済みのモデルをロード
+model = tf.keras.applications.MobileNetV2(weights='imagenet')
+
+# 画像の前処理を行う関数
+def preprocess_image(image):
+    image = image.resize((224, 224))
+    image = np.array(image)
+    image = np.expand_dims(image, axis=0)
+    image = tf.keras.applications.mobilenet_v2.preprocess_input(image)
+    return image
+
+# 予測を行う関数
+def predict(image):
+    processed_image = preprocess_image(image)
+    predictions = model.predict(processed_image)
+    decoded_predictions = tf.keras.applications.mobilenet_v2.decode_predictions(predictions, top=1)
+    return decoded_predictions[0][0][1]
+
+# Streamlitの設定
+st.title("Image Classification with CNN")
+st.write("Upload an image to classify it using MobileNetV2")
+
+# ファイルアップロード
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    image = Image.open(uploaded_file)
+    st.image(image, caption='Uploaded Image.', use_column_width=True)
+    st.write("")
+    st.write("Classifying...")
+    label = predict(image)
+    st.write(f"Prediction: {label}")
+

requirements.txt

@@ -0,0 +1,4 @@
+streamlit
+tensorflow
+Pillow
+