Create app.py

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+from huggingface_hub import hf_hub_download
+import tensorflow as tf
+
+# Hugging Face Hub에서 모델 파일 다운로드
+model_path = hf_hub_download(repo_id="safoinme/zenml-mnist", filename="model.h5")
+
+# 다운로드한 모델 파일로부터 Keras 모델 로드
+model = tf.keras.models.load_model(model_path)
+
+def recognize_digit(img):
+    # 이미지 전처리
+    img = img.reshape(1, 28, 28, 1) / 255.0
+    # 모델 예측
+    prediction = model.predict(img).argmax()
+    return str(prediction)
+
+# Gradio 인터페이스 설정
+iface = gr.Interface(fn=recognize_digit, 
+                     inputs=gr.inputs.Image(shape=(28,28), image_mode='L', invert_colors=True, source="canvas"), 
+                     outputs="text", 
+                     title="MNIST Digit Recognizer",
+                     description="Draw a digit on the canvas and click 'Submit' to recognize it. Model trained on the MNIST dataset.")
+# 인터페이스 실행
+iface.launch()