Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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))
|