Update app.py

app.py

@@ -40,12 +40,12 @@ def inference(raw_image, input_tag):
     with torch.no_grad():
 
 
-        caption, tag_predict = model.generate(image,tag_input = input_tag_list, return_tag_predict = True)
+        caption, tag_predict = model.generate(image,tag_input = input_tag_list,max_length = 50, return_tag_predict = True)
         if input_tag_list == None:
             tag_1 = tag_predict
             tag_2 = ['none']
         else:
-            _, tag_1 = model.generate(image,tag_input = None, return_tag_predict = True)
+            _, tag_1 = model.generate(image,tag_input = None, max_length = 50, return_tag_predict = True)
             tag_2 = tag_predict
 
         return tag_1[0],tag_2[0],caption[0]