Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,11 +36,6 @@ cassio.init(token=ASTRA_DB_APPLICATION_TOKEN, database_id=ASTRA_DB_ID)
|
|
| 36 |
# Initialize LLM & Embeddings
|
| 37 |
login(token=HUGGINGFACE_API_KEY)
|
| 38 |
|
| 39 |
-
hf_llm = HuggingFaceHub(
|
| 40 |
-
repo_id="google/flan-t5-large",
|
| 41 |
-
model_kwargs={"temperature": 0, "max_length": 64, "task": "text2text-generation"},
|
| 42 |
-
huggingfacehub_api_token=HUGGINGFACE_API_KEY
|
| 43 |
-
)
|
| 44 |
|
| 45 |
embedding = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
|
| 46 |
|
|
|
|
| 36 |
# Initialize LLM & Embeddings
|
| 37 |
login(token=HUGGINGFACE_API_KEY)
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
|
| 40 |
embedding = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
|
| 41 |
|