Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
|
|
7 |
from langchain.prompts import ChatPromptTemplate
|
8 |
from langchain.chains import LLMChain
|
9 |
from langchain.llms import CTransformers
|
|
|
10 |
import torch
|
11 |
|
12 |
# ==== Configuration ====
|
@@ -41,9 +42,7 @@ embedding_model = HuggingFaceEmbeddings(
|
|
41 |
# model_type='llama',
|
42 |
# config={'max_new_tokens': 512, 'temperature': 0.4}
|
43 |
# )
|
44 |
-
from langchain.llms import HuggingFaceHub
|
45 |
|
46 |
-
hf_token = "your_huggingface_token"
|
47 |
|
48 |
llm = HuggingFaceHub(
|
49 |
repo_id="epfl-llm/meditron-7b", # Or BioGPT, GatorTron, ClinicalT5, etc.
|
|
|
7 |
from langchain.prompts import ChatPromptTemplate
|
8 |
from langchain.chains import LLMChain
|
9 |
from langchain.llms import CTransformers
|
10 |
+
from langchain.llms import HuggingFaceHub
|
11 |
import torch
|
12 |
|
13 |
# ==== Configuration ====
|
|
|
42 |
# model_type='llama',
|
43 |
# config={'max_new_tokens': 512, 'temperature': 0.4}
|
44 |
# )
|
|
|
45 |
|
|
|
46 |
|
47 |
llm = HuggingFaceHub(
|
48 |
repo_id="epfl-llm/meditron-7b", # Or BioGPT, GatorTron, ClinicalT5, etc.
|