Update app.py
Browse files
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-
|
| 16 |
-
model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-
|
| 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):
|