wakeupmh commited on
Commit
cf78fb9
·
1 Parent(s): 3dcd44f

fix: using falcon instead

Browse files
Files changed (1) hide show
  1. services/model_handler.py +1 -1
services/model_handler.py CHANGED
@@ -1,5 +1,5 @@
1
  import logging
2
- from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import streamlit as st
4
  from agno.agent import Agent
5
  from agno.tools.arxiv import ArxivTools
 
1
  import logging
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
  import streamlit as st
4
  from agno.agent import Agent
5
  from agno.tools.arxiv import ArxivTools