gufett0 commited on
Commit
8acde95
·
1 Parent(s): 29f4899

switched back to langchain

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. backend.py +1 -1
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import os
 
2
  from threading import Thread
3
  from typing import Iterator
4
  from backend2 import load_documents, prepare_documents, get_context_sources
5
  import gradio as gr
6
- import spaces
7
  import torch
8
  from transformers import AutoModelForCausalLM, GemmaTokenizerFast, TextIteratorStreamer
9
  from huggingface_hub import login
 
1
  import os
2
+ import spaces
3
  from threading import Thread
4
  from typing import Iterator
5
  from backend2 import load_documents, prepare_documents, get_context_sources
6
  import gradio as gr
 
7
  import torch
8
  from transformers import AutoModelForCausalLM, GemmaTokenizerFast, TextIteratorStreamer
9
  from huggingface_hub import login
backend.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
 
2
  import os
3
  from transformers import AutoModelForCausalLM, GemmaTokenizerFast, TextIteratorStreamer, AutoTokenizer
4
  from interface import GemmaLLMInterface
@@ -7,7 +8,6 @@ from llama_index.embeddings.instructor import InstructorEmbedding
7
  import gradio as gr
8
  from llama_index.core import Settings, VectorStoreIndex, SimpleDirectoryReader, ChatPromptTemplate, PromptTemplate, load_index_from_storage, StorageContext
9
  from llama_index.core.node_parser import SentenceSplitter
10
- import spaces
11
  from huggingface_hub import login
12
  from llama_index.core.memory import ChatMemoryBuffer
13
  from typing import Iterator, List, Any
 
1
  import torch
2
+ import spaces
3
  import os
4
  from transformers import AutoModelForCausalLM, GemmaTokenizerFast, TextIteratorStreamer, AutoTokenizer
5
  from interface import GemmaLLMInterface
 
8
  import gradio as gr
9
  from llama_index.core import Settings, VectorStoreIndex, SimpleDirectoryReader, ChatPromptTemplate, PromptTemplate, load_index_from_storage, StorageContext
10
  from llama_index.core.node_parser import SentenceSplitter
 
11
  from huggingface_hub import login
12
  from llama_index.core.memory import ChatMemoryBuffer
13
  from typing import Iterator, List, Any