fffiloni commited on
Commit
64e42e3
·
verified ·
1 Parent(s): 06e900c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -102,10 +102,14 @@ Here's another example. If a user types, "In the image, there is a drawing of a
102
 
103
  def infer(image_in, cap_type):
104
  gr.Info("Getting image description...")
 
105
  if cap_type == "Fictional" :
106
  user_prompt = get_caption_from_MD(image_in)
107
  elif cap_type == "Literal" :
108
  user_prompt = get_caption_from_kosmos(image_in)
 
 
 
109
 
110
  gr.Info("Building a system according to the image caption ...")
111
  outputs = get_llm_idea(user_prompt)
 
102
 
103
  def infer(image_in, cap_type):
104
  gr.Info("Getting image description...")
105
+ """
106
  if cap_type == "Fictional" :
107
  user_prompt = get_caption_from_MD(image_in)
108
  elif cap_type == "Literal" :
109
  user_prompt = get_caption_from_kosmos(image_in)
110
+ """
111
+ user_prompt = get_caption_from_kosmos(image_in)
112
+
113
 
114
  gr.Info("Building a system according to the image caption ...")
115
  outputs = get_llm_idea(user_prompt)