multimodalart HF staff commited on
Commit
bfef560
1 Parent(s): 0d76136

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -71,11 +71,6 @@ def load_captioning(uploaded_images, concept_sentence):
71
  updates.append(gr.update(placeholder=f"A {concept_sentence} in a mall"))
72
  return updates
73
 
74
-
75
- if is_spaces:
76
- load_captioning = spaces.GPU()(load_captioning)
77
-
78
-
79
  def create_dataset(*inputs):
80
  print("Creating dataset")
81
  images = inputs[0]
@@ -133,6 +128,8 @@ def run_captioning(images, concept_sentence, *captions):
133
  del model
134
  del processor
135
 
 
 
136
 
137
  def start_training(
138
  profile: Union[gr.OAuthProfile, None],
 
71
  updates.append(gr.update(placeholder=f"A {concept_sentence} in a mall"))
72
  return updates
73
 
 
 
 
 
 
74
  def create_dataset(*inputs):
75
  print("Creating dataset")
76
  images = inputs[0]
 
128
  del model
129
  del processor
130
 
131
+ if is_spaces:
132
+ load_captioning = spaces.GPU()(run_captioning)
133
 
134
  def start_training(
135
  profile: Union[gr.OAuthProfile, None],