test

app.py

@@ -157,7 +157,7 @@ def stop_now():
 
 def main():
     parser = argparse.ArgumentParser(description="CogAgent Gradio Demo")
-    parser.add_argument("--model_dir", default="THUDM/cogagent1.5-9b", help="Path or identifier of the model.")
+    parser.add_argument("--model_dir", default="THUDM/cogagent-9b-20241220", help="Path or identifier of the model.")
     parser.add_argument("--format_key", default="action_op_sensitive", help="Key to select the prompt format.")
     parser.add_argument("--platform", default="Mac", help="Platform information string.")
     parser.add_argument("--output_dir", default="outputs", help="Directory to save annotated images.")
@@ -184,7 +184,7 @@ def main():
     format_str = format_dict[args.format_key]
 
     with gr.Blocks(analytics_enabled=False) as demo:
-        gr.HTML("<h1 align='center'>CogAgent1.5-9B Demo</h1>")
+        gr.HTML("<h1 align='center'>CogAgent-9B-20241220 Demo</h1>")
         gr.HTML(
             "<p align='center' style='color:red;'>This Demo is for learning and communication purposes only. Users must assume responsibility for the risks associated with AI-generated planning and operations.</p>")
 

requirements.txt

@@ -1,19 +1,19 @@
-transformers>=4.47.0
+transformers==4.47.0
 torch==2.2.0
 torchvision
-huggingface-hub>=0.25.1
-sentencepiece>=0.2.0
-jinja2>=3.1.4
-pydantic>=2.9.2
-timm>=1.0.9
-tiktoken>=0.8.0
+huggingface-hub==0.25.1
+sentencepiece==0.2.0
+jinja2==3.1.4
+pydantic==2.9.2
+timm==1.0.9
+tiktoken==0.8.0
 numpy==1.26.4
-accelerate>=1.1.1
-sentence_transformers>=3.1.1
-gradio>=5.9.0
-openai>=1.57.0
-einops>=0.8.0
-pillow>=10.4.0
-sse-starlette>=2.1.3
+accelerate==1.1.1
+sentence_transformers==3.1.1
+gradio==5.9.0
+openai==1.57.0
+einops==0.8.0
+pillow==10.4.0
+sse-starlette==2.1.3
 bitsandbytes>=0.43.2
-spaces>=0.31.1
+spaces==0.31.1