Update models.py

models.py

@@ -1,6 +1,4 @@
-import torch
-
-predictions = torch.tensor([117.3959, 115.1554,  77.0136, 110.3882,  88.8630, 121.9282, 107.0176,
+predictions = [117.3959, 115.1554,  77.0136, 110.3882,  88.8630, 121.9282, 107.0176,
         96.6535, 142.0282,  96.6855, 119.4232, 102.9950,  95.6350, 140.3941,
     120.6821, 128.7494,  50.8274,  45.3665, 130.4926,  21.1215,  98.4367,
     119.0188, 142.0412,  14.2590, 150.3219, 127.5539,  83.7680, 105.2201,
@@ -14,9 +12,9 @@ predictions = torch.tensor([117.3959, 115.1554,  77.0136, 110.3882,  88.8630, 12
     185.3060, 124.4552,  88.0197,   9.2142,  18.2898, 155.0038,  67.7192,
     157.0159, 150.3746, 128.8048, 122.3546, 111.5525,  32.0154,   8.8541,
         23.5993,  23.7085,  63.4035, 155.8032, 112.4585, 117.9760,  77.4710,
-    152.4733,  18.4411])
+    152.4733,  18.4411]
 
-truths = torch.tensor([112.,  98.,  69.,  82.,  91.,  93.,  91.,  95., 111.,  96.,  97., 124.,
+truths = [112.,  98.,  69.,  82.,  91.,  93.,  91.,  95., 111.,  96.,  97., 124.,
         95., 107.,  83.,  84.,  50.,  28.,  87.,  16.,  57., 111., 113.,  20.,
     145., 119.,  66.,  97.,  90., 115.,   8.,  48., 106.,   7.,  11.,  19.,
         21.,  50., 142.,  28.,  18.,  10.,  59., 109., 114.,  47., 135.,  92.,
@@ -24,7 +22,7 @@ truths = torch.tensor([112.,  98.,  69.,  82.,  91.,  93.,  91.,  95., 111.,  96
         21.,  54.,  72.,  28., 128.,  14.,  77.,   8., 121.,  94., 118.,  50.,
     131., 126., 113.,  10.,  34., 107.,  63.,  90.,   8.,   9., 137.,  58.,
     118.,  89., 116., 115., 136.,  28.,  38.,  20.,  85.,  55., 128., 137.,
-        82.,  59., 117.,  20.])
+        82.,  59., 117.,  20.]
 mse = 520.23
 l1 = 16.26