Spaces:

Edgar404
/

Candy_Prototype

Sleeping

Edgar404 commited on Jun 7

Commit

f93c354

•

1 Parent(s): 9fa69d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def process_image(image , mode = 'optimized' ):
     model = optimized_model if mode == 'optimized' else base_model
     processor = optimized_processor if mode == 'optimized' else base_processor
-    d_type = torch.bfloat16 if (mode == 'optimized' & device =='cuda') else torch.float32
     model.to(device)
     model.eval()

     model = optimized_model if mode == 'optimized' else base_model
     processor = optimized_processor if mode == 'optimized' else base_processor
+    d_type = torch.bfloat16 if ((mode == 'optimized') & (device =='cuda')) else torch.float32
     model.to(device)
     model.eval()