Hugging Face's logo

Spaces:
Rahul-8799
/
MAC
Runtime error

App Files Community
Rahul-8799 commited on
Commit
2e22d6c
·
verified ·
1 Parent(s): cc2b268

Update agents/quality_assurance_agent.py

Browse files
Files changed (1) hide show
  1. agents/quality_assurance_agent.py +1 -0
agents/quality_assurance_agent.py CHANGED
@@ -1,5 +1,6 @@
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
 
3
 
4
  MODEL_REPO = "Rahul-8799/quality_assurance_stablecode"
5
 
 
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
3
+ from langchain_core.messages import AIMessage
4
 
5
  MODEL_REPO = "Rahul-8799/quality_assurance_stablecode"
6