# gradio.app

**Repository Path**: liuwenjing2/gradio.app

## Basic Information

- **Project Name**: gradio.app
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-04-10
- **Last Updated**: 2024-04-10

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 项目描述

这是一个基于Python的机器学习项目，TensorFlow和Gradio的机器学习项目。项目包括了模型的训练和推理功能。使用了MNIST手写数字数据集进行训练和测试。

`model training and saving.py`：定义了一个MLP模型，模型的参数会被保存为`best_model.h5`。

`gradio app.py`：实现了一个基于Gradio的图形用户界面（GUI），用于手写数字的识别。它加载了一个预训练的模型（保存为best_model.h5文件），通过用户在GUI中绘制的手写数字图像，使用该模型进行预测，并将预测结果显示在界面上。

## 项目运行效果截图
![输入图片说明](1.jpg)
![输入图片说明](2.JPG)
## 功能

- `train_and_save_models.py`：训练模型并保存。
- `gradio app.py`：加载模型进行预测，并将预测结果显示在界面上

## 依赖

- Python 3
- PyTorch

## 使用

1. 运行`train_and_save_models.py`训练模型并保存。
2. 运行`gradio app.py`加载模型并进行推理。

## 注意

确保在运行`gradio app.py`之前已经运行过`train_and_save_models.py`并成功保存了模型。

## 个人信息


- 学号: 202152320124
- 年级: 2021
- 专业: 智能科学与技术
- 班级: 一班