Update app.py

app.py

@@ -17,7 +17,10 @@ import secrets
 valid_api_keys = set()
 
 def generate_api_key():
-    """Generate a secure 32-character hexadecimal API key and store it."""
+    """
+    Generate a secure 32-character hexadecimal API key,
+    store it in the valid_api_keys set, and return it.
+    """
     key = secrets.token_hex(16)
     valid_api_keys.add(key)
     return key
@@ -49,7 +52,9 @@ def llama_vision_inference(api_key, image):
 
 with gr.Blocks(title="Manus 🚀") as demo:
     gr.Markdown("# Manus 🚀")
-    gr.Markdown("This Gradio Space lets you generate an API key and perform vision inference using the Llama Vision model (running in CPU mode).")
+    gr.Markdown(
+        "This Gradio Space lets you generate an API key and perform vision inference using the Llama Vision model (running in CPU mode)."
+    )
 
     with gr.Tabs():
         with gr.TabItem("API Key Generator"):
@@ -59,7 +64,9 @@ with gr.Blocks(title="Manus 🚀") as demo:
             generate_button.click(fn=generate_api_key, outputs=key_output)
         
         with gr.TabItem("Vision Model Inference"):
-            gr.Markdown("Enter your API key and upload an image (e.g., a Linux GUI screenshot) to run inference using the Llama Vision model.")
+            gr.Markdown(
+                "Enter your API key and upload an image (e.g., a Linux GUI screenshot) to run inference using the Llama Vision model."
+            )
             api_key_input = gr.Textbox(label="API Key", placeholder="Enter your API key here")
             image_input = gr.Image(type="pil", label="Upload Image")
             output_text = gr.Textbox(label="Model Output", interactive=False)
@@ -67,5 +74,4 @@ with gr.Blocks(title="Manus 🚀") as demo:
             run_button.click(fn=llama_vision_inference, inputs=[api_key_input, image_input], outputs=output_text)
 
 if __name__ == "__main__":
-    # Disable SSR to avoid potential GPU-related initialization issues.
-    demo.launch(ssr=False)
+    demo.launch()