Update main.py

main.py

@@ -30,10 +30,6 @@ def read_root():
 def read_root(data: req):
   print("Prompt:", data.prompt)
   print("Length:", data.length)
-
-  # Using CUDA for an optimal experience
-  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-  model = model.to(device)
   
   input_text = data.prompt