marii8st commited on
Commit
7f0a050
·
verified ·
1 Parent(s): a712d9a

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -32
app.py DELETED
@@ -1,32 +0,0 @@
1
- import gradio as gr
2
- from tensorflow.keras.models import load_model
3
- from tensorflow.keras.preprocessing.image import img_to_array
4
- from PIL import Image
5
- import numpy as np
6
-
7
- # Загрузка модели
8
- model = load_model("digit_recognizer_model.h5")
9
-
10
- # Функция для обработки изображения и предсказания
11
- def predict_digit(image):
12
- image = image.convert("L") # Конвертация в градации серого
13
- image = image.resize((28, 28)) # Изменение размера на 28x28
14
- image = img_to_array(image) # Преобразование в numpy-массив
15
- image = image.reshape(1, 28, 28, 1) # Преобразование в формат (1, 28, 28, 1)
16
- image = image / 255.0 # Нормализация
17
-
18
- prediction = model.predict(image) # Предсказание
19
- predicted_class = np.argmax(prediction) # Получение класса с наибольшей вероятностью
20
- return f"Предсказанное число: {predicted_class}"
21
-
22
- # Интерфейс Gradio
23
- interface = gr.Interface(
24
- fn=predict_digit,
25
- inputs=gr.Image(type="pil", shape=(28, 28)), # Новый формат ввода изображения
26
- outputs="text", # Вывод в виде текста
27
- title="Распознавание цифр"
28
- )
29
-
30
- # Запуск приложения
31
- if __name__ == "__main__":
32
- interface.launch()