LuisMBA commited on
Commit
b5056e8
·
verified ·
1 Parent(s): 98f5046

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  from logic import get_author_by_title, get_year_by_title, get_summary_by_title, find_book_by_description
4
 
5
  # Load the fine-tuned model
6
- model_name = "./fine_tuned_model"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9
 
 
3
  from logic import get_author_by_title, get_year_by_title, get_summary_by_title, find_book_by_description
4
 
5
  # Load the fine-tuned model
6
+ model_name = "LuisMBA/sci_fi_books_chat_100_ex"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  model = AutoModelForCausalLM.from_pretrained(model_name)
9