Update app.py
Browse files
app.py
CHANGED
@@ -91,9 +91,7 @@ class ConversationBot:
|
|
91 |
def __init__(self, load_dict):
|
92 |
# load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
|
93 |
print(f"Initializing VisualChatGPT, load_dict={load_dict}")
|
94 |
-
|
95 |
-
raise ValueError("You have to load ImageCaptioning as a basic function for VisualChatGPT")
|
96 |
-
|
97 |
self.models = {}
|
98 |
# Load Basic Foundation Models
|
99 |
for class_name, device in load_dict.items():
|
|
|
91 |
def __init__(self, load_dict):
|
92 |
# load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
|
93 |
print(f"Initializing VisualChatGPT, load_dict={load_dict}")
|
94 |
+
|
|
|
|
|
95 |
self.models = {}
|
96 |
# Load Basic Foundation Models
|
97 |
for class_name, device in load_dict.items():
|