Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -41,9 +41,7 @@ class ModelManager:
|
|
41 |
)
|
42 |
|
43 |
self.model = Blip2ForConditionalGeneration.from_pretrained("banao-tech/OmniParser",torch_dtype=torch.float16,
|
44 |
-
trust_remote_code=True)
|
45 |
-
|
46 |
-
).to(self.device)
|
47 |
|
48 |
return True
|
49 |
except Exception as e:
|
|
|
41 |
)
|
42 |
|
43 |
self.model = Blip2ForConditionalGeneration.from_pretrained("banao-tech/OmniParser",torch_dtype=torch.float16,
|
44 |
+
trust_remote_code=True).to(self.device)
|
|
|
|
|
45 |
|
46 |
return True
|
47 |
except Exception as e:
|