Update app.py

app.py

@@ -12,7 +12,8 @@ from util import *
 import gradio as gr
 from chatbot import *
 from load_internvideo import *
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+device = torch.device('cuda'
+ if torch.cuda.is_available() else 'cpu')
 from simplet5 import SimpleT5
 from models.grit_model import DenseCaptioning
 bot = ConversationBot()
@@ -42,7 +43,7 @@ trans_action = transform_action()
 topil =  T.ToPILImage()
 print("[INFO] initialize InternVideo model success!")
 
-dense_caption_model = DenseCaptioning(device)
+dense_caption_model = DenseCaptioning(str(device))
 dense_caption_model.initialize_model()
 print("[INFO] initialize dense caption model success!")