interface.py

import streamlit as st
import torch
from torchvision.utils import make_grid
from torchvision.transforms import ToPILImage


def main():
    st.title("Image Generation")
    st.write("Made with GANS from scratch")


if __name__ == "__main__":
    main()