upd

src/streamlit_app.py

@@ -15,7 +15,7 @@ def load_model():
     cache_dir = "/tmp/model_cache"
     os.makedirs(cache_dir, exist_ok=True)
     
-    tokenizer = AutoTokenizer.from_pretrained("tianzhechu/BookQA-7B-Instruct", cache_dir=cache_dir)
+    tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-0.5B-Instruct", cache_dir=cache_dir)
     model = AutoModelForSeq2SeqLM.from_pretrained("tianzhechu/BookQA-7B-Instruct", cache_dir=cache_dir)
     return pipeline("text2text-generation", model=model, tokenizer=tokenizer)