shuangzhiaishang commited on
Commit
39eba3e
·
verified ·
1 Parent(s): 16b2893

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -7,13 +7,14 @@ from model import ToyModel
7
  """
8
  Model specification
9
  """
10
-
11
- model = ToyModel()
 
12
 
13
 
14
  def chat(image_input, text_input):
15
- image_output, text_output = model.chat(image_input, text_input)
16
- return image_output, text_output
17
 
18
 
19
  """
 
7
  """
8
  Model specification
9
  """
10
+ vision_model_path = 'clip-vit-base-patch32'
11
+ language_model_path = 'gpt2'
12
+ model = ToyModel(vision_model_path, language_model_path)
13
 
14
 
15
  def chat(image_input, text_input):
16
+ text_output = model.chat(image_input, text_input)
17
+ return image_input, text_output
18
 
19
 
20
  """