Skip to content

Translation from English to Spanish, all the way from tokenisation to beam search

Notifications You must be signed in to change notification settings

Vanekkk1/transformer_from_scratch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

English-Spanish Translation Model

This project focuses on building a machine learning model capable of translating English text into Spanish. It utilizes TensorFlow and Keras for model development and incorporates Streamlit for web application integration.

Usage

  • Model Script: The main model is located in app/model.py.
  • Model and App Utilities: Various helpers for both the model and the app can be found in app/model_utils.py.
  • Streamlit Integration: The Streamlit app is defined in app/app.py.
  • Text Vectorization: The script vectorization.py is utilized for text vectorization (tokenization).
  • Trained Model: The trained model is saved as eng_esp_model.keras.
  • Model Testing: The notebook eng_spanish_translation.ipynb is used for testing the model.

Contributing

Contributions to this project are highly encouraged! If you would like to contribute, please submit a pull request.

License

This project is licensed under the MIT License, allowing for wide use and distribution.

Deployment

The model is deployed at the following URL: https://transformermodel.streamlit.app

Note: The deployment might not be operational due to limitations of the free tier on Streamlit.

About

Translation from English to Spanish, all the way from tokenisation to beam search

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published