nurqoneah commited on
Commit
87375d5
·
verified ·
1 Parent(s): 4d75c2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,8 @@ from langchain.prompts import PromptTemplate
6
  from langchain.chains import RetrievalQA, ConversationalRetrievalChain
7
  from langchain.memory import ConversationBufferMemory
8
  import warnings
 
 
9
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
10
  import os
11
  from dotenv import load_dotenv
 
6
  from langchain.chains import RetrievalQA, ConversationalRetrievalChain
7
  from langchain.memory import ConversationBufferMemory
8
  import warnings
9
+ from transformers import pipeline
10
+ import torch
11
  from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
12
  import os
13
  from dotenv import load_dotenv