Update app.py

app.py

@@ -13,9 +13,10 @@ from langchain.prompts import PromptTemplate
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_chroma import Chroma
 from langchain_community.vectorstores import Chroma
-from transformers import AutoTokenizer, AutoModelForCausalLM
-import transformers
-import torch
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+
 
 #st.set_page_config(page_title="Document Genie", layout="wide")
 
@@ -89,11 +90,7 @@ def get_conversational_chain():
     model_id = "google/gemma-1.1-2b-it"
     dtype = torch.bfloat16
 
-    tokenizer = AutoTokenizer.from_pretrained(model_id)
-    llm= AutoModelForCausalLM.from_pretrained(
-    model_id,
-    torch_dtype=dtype,
-    )
+   llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
 
     pt = ChatPromptTemplate.from_template(template)
     # Retrieve and generate using the relevant snippets of the blog.