Spaces:

Rahul-8799
/

Multi_agent_test

Runtime error

Rahul-8799 commited on May 10

Commit

1bc941f

verified ·

1 Parent(s): a640c55

Update agents/ui_designer_agent.py

Files changed (1) hide show

agents/ui_designer_agent.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 from langchain_core.messages import AIMessage
-MODEL_REPO = "Rahul-8799/ui_designer_mistral"  # You'll need to fine-tune this model
 tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(

 import torch
 from langchain_core.messages import AIMessage
+MODEL_REPO = "Rahul-8799/product_manager_mistral"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(