Spaces:

ahmed-masry
/

ColFlor-Demo

Sleeping

ahmed-masry commited on Oct 8, 2024

Commit

979a12b

verified ·

1 Parent(s): 47f0faa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,6 @@ token = os.environ.get("HF_TOKEN")
 model = ColFlor.from_pretrained(
     model_name, torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
-model = model.eval()
 processor = ColFlorProcessor.from_pretrained(model_name, token = token)
 mock_image = Image.new("RGB", (768, 768), (255, 255, 255))

 model = ColFlor.from_pretrained(
     model_name, torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
 processor = ColFlorProcessor.from_pretrained(model_name, token = token)
 mock_image = Image.new("RGB", (768, 768), (255, 255, 255))