rishisim commited on
Commit
9adf0de
1 Parent(s): d64797e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -2,10 +2,13 @@ import gradio as gr
2
  # from langchain.llms import GooglePalm
3
  from langchain_google_genai import GoogleGenerativeAI
4
 
 
 
 
5
 
6
  api_key = "AIzaSyCdM_aAIsW_nPbjarOF83mbX1_z1cVX2_M"
7
 
8
- llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)
9
  # llm = GooglePalm(google_api_key = api_key, temperature=0.7)
10
 
11
  from langchain.document_loaders.csv_loader import CSVLoader
 
2
  # from langchain.llms import GooglePalm
3
  from langchain_google_genai import GoogleGenerativeAI
4
 
5
+ from transformers import AutoModelForCausalLM, AutoTokenizer
6
+
7
+ llm = "mistralai/Mixtral-8x22B-Instruct-v0.1"
8
 
9
  api_key = "AIzaSyCdM_aAIsW_nPbjarOF83mbX1_z1cVX2_M"
10
 
11
+ # llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)
12
  # llm = GooglePalm(google_api_key = api_key, temperature=0.7)
13
 
14
  from langchain.document_loaders.csv_loader import CSVLoader