Upload app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import os
 from pathlib import Path
 import fitz  # PyMuPDF
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import Chroma
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import Chroma
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.chains import RetrievalQA
-from langchain.llms import HuggingFacePipeline
+from langchain_community.llms import HuggingFacePipeline
 from langchain.prompts import PromptTemplate
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
@@ -30,7 +30,7 @@ class CurriculumAssistant:
             model = AutoModelForCausalLM.from_pretrained(
                 model_name,
                 torch_dtype=torch.float16,
-                device_map="auto",
+                device_map=None,  # Use CPU for Hugging Face Spaces
                 trust_remote_code=True
             )