Update app.py

app.py

@@ -1,6 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
+import torch
+
+zero = torch.Tensor([0]).cuda()
+print(zero.device)
+
 # Code Generation Pipeline
 code_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=0)