Update app.py

app.py

@@ -22,11 +22,7 @@ def getText(filename):
     return '\n'.join(fullText)
     
 
-if torch.cuda.is_available():  
-  dev = "cuda"
-else:  
-  dev = "cpu" 
-device = torch.device(dev)
+
  
 # mname = 'Helsinki-NLP/opus-mt-en-hi'
 # tokenizer = MarianTokenizer.from_pretrained(mname)
@@ -35,6 +31,11 @@ device = torch.device(dev)
 
 #@st.cache
 def btTranslator(docxfile):
+  if torch.cuda.is_available():  
+    dev = "cuda"
+  else:  
+    dev = "cpu" 
+  device = torch.device(dev)
   a=getText(docxfile)
   a1=a.split('\n')
   bigtext='''  '''