k3ybladewielder commited on
Commit
a32972c
β€’
1 Parent(s): d0ab9cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,5 +1,3 @@
1
- !pip install git+https://github.com/huggingface/transformers
2
-
3
  from langchain import HuggingFacePipeline
4
  from langchain import PromptTemplate, LLMChain
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -9,6 +7,11 @@ import os
9
  import torch
10
  import gradio as gr
11
 
 
 
 
 
 
12
  # check if cuda is available
13
  torch.cuda.is_available()
14
 
 
 
 
1
  from langchain import HuggingFacePipeline
2
  from langchain import PromptTemplate, LLMChain
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
7
  import torch
8
  import gradio as gr
9
 
10
+ import subprocess
11
+
12
+ command = 'pip install git+https://github.com/huggingface/transformers'
13
+ subprocess.run(command, shell=True)
14
+
15
  # check if cuda is available
16
  torch.cuda.is_available()
17