Docfile commited on
Commit
155e6d5
·
1 Parent(s): 8e22a1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -1
app.py CHANGED
@@ -1,9 +1,17 @@
1
  import gradio as gr
2
  import os
 
 
 
 
3
 
4
  GOOGLE_API_KEY = "AIzaSyBbruzn10nez-0a-_60TA9R9h6qumLD1Es" # add your GOOGLE API key here
5
  os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
6
 
 
 
 
 
7
 
8
  from llama_index import (
9
  ServiceContext,
@@ -14,11 +22,21 @@ from llama_index import (
14
  )
15
  from llama_index.llms import Gemini
16
  from llama_index.embeddings import GeminiEmbedding
 
 
 
 
 
 
 
 
 
 
17
 
18
  model_name = "models/embedding-001"
19
 
20
 
21
- llm = Gemini()
22
  embed_model = GeminiEmbedding(
23
  model_name=model_name, api_key=GOOGLE_API_KEY, title="this is a document"
24
  )
 
1
  import gradio as gr
2
  import os
3
+ import logging
4
+
5
+ logging.basicConfig(stream=sys.stdout, level=logging.INFO)
6
+ logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
7
 
8
  GOOGLE_API_KEY = "AIzaSyBbruzn10nez-0a-_60TA9R9h6qumLD1Es" # add your GOOGLE API key here
9
  os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
10
 
11
+ from g4f import Provider, models
12
+ from langchain.llms.base import LLM
13
+ from llama_index.llms import LangChainLLM
14
+ from langchain_g4f import G4FLLM
15
 
16
  from llama_index import (
17
  ServiceContext,
 
22
  )
23
  from llama_index.llms import Gemini
24
  from llama_index.embeddings import GeminiEmbedding
25
+ import g4f
26
+ g4f.debug.logging = True
27
+
28
+
29
+ llm= LLM = G4FLLM(
30
+ model=models.gpt_35_turbo_16k,
31
+ ,
32
+ )
33
+
34
+ llm = LangChainLLM(llm=llm)
35
 
36
  model_name = "models/embedding-001"
37
 
38
 
39
+ #llm = Gemini()
40
  embed_model = GeminiEmbedding(
41
  model_name=model_name, api_key=GOOGLE_API_KEY, title="this is a document"
42
  )