Hemasagar commited on
Commit
fd49903
·
verified ·
1 Parent(s): c51fa22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,8 +1,12 @@
1
  import streamlit as st
2
  from langchain.prompts import PromptTemplate
3
  # Recently the below import has been replaced by later one
4
- # from langchain.llms import CTransformers
5
- from langchain_community.llms import CTransformers
 
 
 
 
6
 
7
  #Function to get the response back
8
  def getLLMResponse(form_input,email_sender,email_recipient,email_style):
 
1
  import streamlit as st
2
  from langchain.prompts import PromptTemplate
3
  # Recently the below import has been replaced by later one
4
+ # Use a pipeline as a high-level helper
5
+ from transformers import pipeline
6
+
7
+ CTransformers = pipeline("text-generation", model="TheBloke/Llama-2-7B-Chat-GGML")
8
+ # # from langchain.llms import CTransformers
9
+ # from langchain_community.llms import CTransformers
10
 
11
  #Function to get the response back
12
  def getLLMResponse(form_input,email_sender,email_recipient,email_style):