JustusI commited on
Commit
6dd10e8
·
verified ·
1 Parent(s): eaff927

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ from langchain_openai import ChatOpenAI
12
  from transformers import AutoTokenizer, AutoModelForCausalLM
13
  from bs4 import BeautifulSoup
14
  import requests
 
15
 
16
  # Function to load vector database
17
  def load_vector_db(zip_file_path, extract_path):
 
12
  from transformers import AutoTokenizer, AutoModelForCausalLM
13
  from bs4 import BeautifulSoup
14
  import requests
15
+ import torch
16
 
17
  # Function to load vector database
18
  def load_vector_db(zip_file_path, extract_path):