Veda0718 commited on
Commit
499d2bb
·
verified ·
1 Parent(s): 9d75af8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,4 +1,8 @@
1
  import os
 
 
 
 
2
  import streamlit as st
3
  import torch
4
  from auto_gptq import AutoGPTQForCausalLM
@@ -12,9 +16,6 @@ from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
12
  from langchain_core.prompts import PromptTemplate
13
  from langchain.chains import RetrievalQA
14
 
15
- os.system('pip install auto_gptq-0.4.1+cu118-cp310-cp310-linux_x86_64.whl')
16
- os.system('sudo apt-get install -y poppler-utils')
17
-
18
  DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
19
 
20
  meta_images = convert_from_path("Medical_Book.pdf", dpi=88)
 
1
  import os
2
+
3
+ os.system('pip install auto_gptq-0.4.1+cu118-cp310-cp310-linux_x86_64.whl')
4
+ os.system('sudo apt-get install -y poppler-utils')
5
+
6
  import streamlit as st
7
  import torch
8
  from auto_gptq import AutoGPTQForCausalLM
 
16
  from langchain_core.prompts import PromptTemplate
17
  from langchain.chains import RetrievalQA
18
 
 
 
 
19
  DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
20
 
21
  meta_images = convert_from_path("Medical_Book.pdf", dpi=88)