Skip to content

thonedra-dev/Ai_Model_Hub

Repository files navigation

πŸ€– AI Model Hub

AI Model Hub Banner

Your Gateway to Machine Learning Excellence

A comprehensive full-stack web application featuring multiple AI/ML models for predictions, classifications, and intelligent analysis.

MIT License Python Flask TensorFlow

Features β€’ Models β€’ Installation β€’ Tech Stackβ€’ Tech Stack β€’ Contact


πŸ“Έ Preview

Main Dashboard

AI Model Hub Dashboard

Model Interfaces

Wine Quality Prediction PDF Summarization


🧠 Available Models

Model Type Task Technology Status
🩺 Cancer Prediction Classification Medical diagnosis prediction RandomForest / XGBoost βœ… Active
πŸš— Car Price Prediction Regression Vehicle valuation RandomForest / XGBoost βœ… Active
🏠 House Price Prediction Regression Real estate valuation RandomForest / XGBoost βœ… Active
πŸŽ“ Student Performance Regression Academic outcome prediction RandomForest / XGBoost βœ… Active
🍷 Wine Quality Assessment Classification Wine quality grading RandomForest / XGBoost βœ… Active
πŸ“§ Email Classification NLP Classification Spam/category detection TensorFlow Transformers βœ… Active
πŸ“„ PDF Summarization NLP Summarization Document summarization BART (facebook/bart-large-cnn) βœ… Pre-trained

πŸš€ Installation

Prerequisites

  • Python 3.9.13
  • pip package manager
  • Virtual environment (recommended)

Step-by-Step Setup

1️⃣ Clone the Repository

git clone https://github.com/yourusername/ai-model-hub.git
cd ai-model-hub

2️⃣ Create Virtual Environment

python -m venv venv

3️⃣ Activate Virtual Environment

# Windows
venv\Scripts\activate

# macOS/Linux
source venv/bin/activate

4️⃣ Install Dependencies

pip install -r requirements.txt

⚠️ Important: Strictly follow the dependency versions specified in requirements.txt to ensure compatibility.

5️⃣ Run the Application

python app.py

6️⃣ Access the Application

Open your browser and navigate to: http://localhost:5000 <=== in here, actually, any available port that your local has.

πŸ“¦ Dependencies

Python==3.9.13
Flask==3.1.2
tensorflow==2.10.1
scikit-learn==1.6.1
pandas==2.3.3
numpy==1.26.4
matplotlib==3.9.4
seaborn==0.13.2
h5py==3.14.0
joblib==1.5.3
transformers==latest

πŸ’‘ Note: These exact versions are required for optimal performance and compatibility.


πŸ’» Tech Stack

Frontend

HTML5 CSS3 JavaScript

Backend

Python Flask

Machine Learning & AI

TensorFlow scikit-learn Pandas NumPy

Visualization

Matplotlib Seaborn


πŸ—οΈ Architecture

AI_Model_Hub/
β”œβ”€β”€ app.py                 # Flask application with route handlers
β”œβ”€β”€ templates/             # HTML templates
β”‚   β”œβ”€β”€ index.html        # Main landing page
β”‚   β”œβ”€β”€ cancer.html       # Cancer prediction interface
β”‚   β”œβ”€β”€ car.html          # Car price prediction
β”‚   β”œβ”€β”€ house.html        # House price prediction
β”‚   β”œβ”€β”€ student.html      # Student performance
β”‚   β”œβ”€β”€ wine.html         # Wine quality assessment
β”‚   β”œβ”€β”€ email.html        # Email classification
β”‚   └── upload_pdf.html   # PDF summarization
β”œβ”€β”€ static/
β”‚   └── images/           # UI screenshots
β”œβ”€β”€ deeplearning_models/  # Trained model files (.pkl, .h5)
β”œβ”€β”€ supervised_models/    # Trained model files (.joblib)
β”œβ”€β”€ requirements.txt      # Python dependencies
└── README.md             # Project documentation

πŸ“ License

This project is licensed under the MIT License - see the LICENSE file for details.

MIT License

Copyright (c) 2024 Thonedra

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files...

πŸ‘¨β€πŸ’» Author

Thonedra

Email Phone

Passionate about AI/ML and building intelligent systems


⭐ Show Your Support

If this project helped you, please consider giving it a ⭐️!


πŸ“Š Project Statistics

Models ML Algorithms Deep Learning Pre--trained


🌟 Built with passion for AI/ML excellence 🌟

Made with ❀️ by Thonedra

About

Multi-model ML web application with Flask - Spam detection, house/car price prediction, cancer diagnosis, PDF summarizer, and more

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors