ahmed-masry commited on
Commit
979a12b
·
verified ·
1 Parent(s): 47f0faa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -22,7 +22,6 @@ token = os.environ.get("HF_TOKEN")
22
  model = ColFlor.from_pretrained(
23
  model_name, torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
24
 
25
- model = model.eval()
26
  processor = ColFlorProcessor.from_pretrained(model_name, token = token)
27
 
28
  mock_image = Image.new("RGB", (768, 768), (255, 255, 255))
 
22
  model = ColFlor.from_pretrained(
23
  model_name, torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
24
 
 
25
  processor = ColFlorProcessor.from_pretrained(model_name, token = token)
26
 
27
  mock_image = Image.new("RGB", (768, 768), (255, 255, 255))