zxcgqq commited on
Commit
51f0ac5
·
1 Parent(s): f4a6fe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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
- if 'ImageCaptioning' not in load_dict:
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():