Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -290,6 +290,7 @@ def captioning(img):
|
|
290 |
output_prompt.append(generated_text)
|
291 |
print(generated_text)
|
292 |
# Loop through prompts array:
|
|
|
293 |
for prompt in prompts_array:
|
294 |
inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
|
295 |
#with torch.no_grad():
|
@@ -328,6 +329,7 @@ def captioning(img):
|
|
328 |
response_text = generated_texta.replace(generated_text, "").strip()
|
329 |
print(f"{response_text}\n")
|
330 |
output_prompt.append(response_text)
|
|
|
331 |
output_prompt = " ".join(output_prompt)
|
332 |
return output_prompt
|
333 |
|
|
|
290 |
output_prompt.append(generated_text)
|
291 |
print(generated_text)
|
292 |
# Loop through prompts array:
|
293 |
+
'''
|
294 |
for prompt in prompts_array:
|
295 |
inputs = processor5(images=img, text=prompt, return_tensors="pt").to('cuda')
|
296 |
#with torch.no_grad():
|
|
|
329 |
response_text = generated_texta.replace(generated_text, "").strip()
|
330 |
print(f"{response_text}\n")
|
331 |
output_prompt.append(response_text)
|
332 |
+
'''
|
333 |
output_prompt = " ".join(output_prompt)
|
334 |
return output_prompt
|
335 |
|