istiak101 commited on
Commit
8724e53
·
verified ·
1 Parent(s): 88b5f09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ def load_resources():
12
  load_dotenv()
13
  huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
14
  subprocess.run(["huggingface-cli", "login", "--token", huggingface_token], capture_output=True)
15
- tokenizer = AutoTokenizer.from_pretrained("istiak101/TinyLlama-1.1B-Finetunedv1.0")
16
- model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-Finetunedv1.0")
17
  return model, tokenizer
18
 
19
  def create_test_prompt(question, context, tokenizer):
 
12
  load_dotenv()
13
  huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
14
  subprocess.run(["huggingface-cli", "login", "--token", huggingface_token], capture_output=True)
15
+ tokenizer = AutoTokenizer.from_pretrained("istiak101/TinyLlama-1.1B-Chat-v0.6-rag-finetunedv1.0")
16
+ model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-Chat-v0.6-rag-finetunedv1.0")
17
  return model, tokenizer
18
 
19
  def create_test_prompt(question, context, tokenizer):