DeepVen commited on
Commit
30fe3f4
1 Parent(s): c4cae3f

Upload 2 files

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. Index.py +1 -1
Dockerfile CHANGED
@@ -24,4 +24,4 @@ WORKDIR $HOME/app
24
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
25
  COPY --chown=user . $HOME/app
26
 
27
- CMD ["uvicorn", "Venkat:app", "--host", "0.0.0.0", "--port", "7860"]
 
24
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
25
  COPY --chown=user . $HOME/app
26
 
27
+ CMD ["uvicorn", "Index:app", "--host", "0.0.0.0", "--port", "7860"]
Index.py CHANGED
@@ -228,7 +228,7 @@ def rag(domain: str, question: str):
228
  # Create extractor instance
229
  #extractor = Extractor(embeddings, "google/flan-t5-base")
230
  #extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF")
231
- extractor = Extractor(embeddings, "google/flan-t5-xl")
232
  # llm = HuggingFaceHub(
233
  # repo_id="google/flan-t5-xxl",
234
  # model_kwargs={"temperature": 1, "max_length": 1000000},
 
228
  # Create extractor instance
229
  #extractor = Extractor(embeddings, "google/flan-t5-base")
230
  #extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF")
231
+ extractor = Extractor(embeddings, "google/flan-t5-base")
232
  # llm = HuggingFaceHub(
233
  # repo_id="google/flan-t5-xxl",
234
  # model_kwargs={"temperature": 1, "max_length": 1000000},