Update app.py

app.py

@@ -84,12 +84,12 @@ def classify_GPU(input_data):
     
     for input in input_data:
     
-        x = torch.tensor([input[:1022]] * number_of_batches, dtype=torch.long, device='cuda')
+        x = torch.tensor([input[:1022]] * number_of_batches, dtype=torch.long, device=DEVICE)
     
         with ctx:
           out = model.generate(x,
                               1,
-                              temperature=0.3,
+                              temperature=0.9,
                               filter_logits_fn=top_k,
                               filter_kwargs={'k': 1},
                               return_prime=False,