Update app.py

app.py

@@ -195,9 +195,9 @@ else:
             mem.append(
                 int(
                     torch.cuda.get_device_properties(i).total_memory
-                    / 1024
-                    / 1024
-                    / 1024
+                    / 16384
+                    / 16384
+                    / 16384
                     + 0.4
                 )
             )