diff --git a/.python-version b/.python-version
deleted file mode 100644
index d2577d9..0000000
--- a/.python-version
+++ /dev/null
@@ -1 +0,0 @@
-3.7.7
diff --git a/README.md b/README.md
index e69de29..0ac3585 100644
--- a/README.md
+++ b/README.md
@@ -0,0 +1 @@
+# DRAMA GENERATOR 
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 7a7cd9f..9234c93 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,6 +16,10 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
 ]
+dependencies = [
+    "torch >= 1.10.0",
+    "numpy>=1.21.5"
+    ]
 
 [project.urls]
 "repository" = "https://git.weber.codes/tom/drama_generator.git"
diff --git a/src/drama_generator/model.py b/src/drama_generator/model.py
index f7c1fca..6cf9c74 100644
--- a/src/drama_generator/model.py
+++ b/src/drama_generator/model.py
@@ -2,7 +2,6 @@
 """ module for training and creating the model """
 import torch
 from torch.utils.data import DataLoader
-from tqdm import tqdm  # type: ignore
 import numpy as np
 from drama_generator.data import ShakespeareDataset, preprocess, get_shakespeare
 
@@ -70,7 +69,7 @@ def train(epochs: int, save_path: str):
         num_workers=2,
         drop_last=True,
     )
-    for e in tqdm(range(epochs)):
+    for e in range(epochs):
         l = train_epoch(gru, loader, optim, loss, device)
         print(f"Epoch: {e}, Loss: {l}")
     torch.save(gru.state_dict(), save_path + f"gru_{epochs+1}epochs.pth")