Spaces:

gabrielchua
/

the-poet

Runtime error

Gabriel C commited on Mar 31

Commit

c75ac23

•

1 Parent(s): 02afc77

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-3
 model.to("cuda:0")
 processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-34b-hf")
-@spaces.GPU(duration=120)
 def generate_caption(image):
     """
     Generate a poem from an image.

 model.to("cuda:0")
 processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-34b-hf")
+@spaces.GPU(duration=240)
 def generate_caption(image):
     """
     Generate a poem from an image.