update app

app.py

@@ -1,13 +1,10 @@
 """SpaceLlama3.1 demo gradio app."""
 
-"""SpaceLlama3.1 demo gradio app."""
-
 import datetime
 import logging
 import os
 
 import gradio as gr
-import requests
 import torch
 import PIL.Image
 from prismatic import load
@@ -46,7 +43,7 @@ def compute(image, prompt, model_location):
 
     # Set device and load the model
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-    vlm = load(model_location)
+    vlm = load(model_location)  # No need to pass the token again
     vlm.to(device, dtype=torch.bfloat16)
 
     # Prepare prompt
@@ -93,19 +90,19 @@ def create_app():
 
         # Button event handlers
         run.click(
-            compute,
-            [image, prompt, model_location],
-            highlighted_text,
+            fn=compute,
+            inputs=[image, prompt, model_location],
+            outputs=highlighted_text,
         )
-        clear.click(reset, None, [prompt, image])
+        clear.click(fn=reset, inputs=None, outputs=[prompt, image])
 
         # Status
         status = gr.Markdown(f"Startup: {datetime.datetime.now()}")
         gpu_kind = gr.Markdown(f"GPU=?")
         demo.load(
-            lambda: [f"Model `{model_location}` loaded."],
-            None,
-            model_info,
+            fn=lambda: [f"Model `{model_location}` loaded."],
+            inputs=None,
+            outputs=model_info,
         )
 
     return demo
@@ -120,3 +117,4 @@ if __name__ == "__main__":
         logging.info('environ["%s"] = %r', k, v)
 
     create_app().queue().launch()
+