Update app.py

app.py

@@ -8,10 +8,10 @@ import os
 token = os.getenv('token')
 HfFolder.save_token(token)
 device = "cuda"
-model = PaliGemmaForConditionalGeneration.from_pretrained("triphuong57/paligemma_lora")
+model = PaliGemmaForConditionalGeneration.from_pretrained("triphuong57/paligemma_lora").to(device)
 processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
 
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=180)
 def greet(image, prompt):
     
     
@@ -26,5 +26,7 @@ def greet(image, prompt):
         decoded = processor.decode(generation[0][input_len:], skip_special_tokens=True)
     return decoded
 
-demo = gr.Interface(fn=greet, inputs=[gr.Image(label="Upload image", sources=['upload', 'webcam'], type="pil"), gr.Text()], outputs="text")
+title = "Demo BTL nhóm 8"
+description = "Made by Nguyễn Quý Đang, Đỗ Minh Nhật, Vũ Vân Long"
+demo = gr.Interface(fn=greet, inputs=[gr.Image(label="Upload image", sources=['upload', 'webcam'], type="pil"), gr.Text()], outputs="text", title=title, description=description)
 demo.launch(share=True)