From 5f36d2af4c2cf7610279475151dbf7bf3f522d43 Mon Sep 17 00:00:00 2001
From: Daniel Kleine <53251018+d-kleine@users.noreply.github.com>
Date: Sun, 22 Sep 2024 18:59:37 +0200
Subject: [PATCH] removed unnecessary imports (#367)

---
 ch06/04_user_interface/app.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/ch06/04_user_interface/app.py b/ch06/04_user_interface/app.py
index 5313b2f..84ad330 100644
--- a/ch06/04_user_interface/app.py
+++ b/ch06/04_user_interface/app.py
@@ -52,7 +52,6 @@ def get_model_and_tokenizer():
     model.out_head = torch.nn.Linear(in_features=GPT_CONFIG_124M["emb_dim"], out_features=num_classes)
 
     # Then load model weights
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     checkpoint = torch.load(model_path, map_location=device, weights_only=True)
     model.load_state_dict(checkpoint)
     model.to(device)
@@ -71,7 +70,6 @@ async def main(message: chainlit.Message):
     The main Chainlit function.
     """
     user_input = message.content
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     label = classify_review(user_input, model, tokenizer, device, max_length=120)