rbughao commited on
Commit
2a0a178
·
verified ·
1 Parent(s): 8d32209

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -1,11 +1,19 @@
1
  #for learning
2
  import os
3
- import openai
4
  import gradio as gr
5
  from llama_index.readers.file import PDFReader
6
 
 
7
 
8
- openai.api_key = os.environ.get('O_APIKey')
 
 
 
 
 
 
 
9
  Data_Read = os.environ.get('Data_Reader')
10
  ChurnData = os.environ.get('Churn_Data')
11
  ChurnData2 = os.environ.get('Churn_Data2')
 
1
  #for learning
2
  import os
3
+ #import openai
4
  import gradio as gr
5
  from llama_index.readers.file import PDFReader
6
 
7
+ ### added to remove openapi
8
 
9
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
10
+ model_name = "baffo32/decapoda-research-llama-7B-hf"
11
+ model = torch.hub.load("HuggingFace", "transformers", model_name).to(device)
12
+
13
+ ### added to remove openapi
14
+
15
+
16
+ #openai.api_key = os.environ.get('O_APIKey')
17
  Data_Read = os.environ.get('Data_Reader')
18
  ChurnData = os.environ.get('Churn_Data')
19
  ChurnData2 = os.environ.get('Churn_Data2')