shethjenil commited on
Commit
56367c9
1 Parent(s): 142978b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,12 +1,10 @@
1
  import chromadb
2
  import requests
3
  import chromadb.utils.embedding_functions as embedding_functions
4
- from ai4bharat.transliteration import XlitEngine
5
  import gradio as gr
6
  import os
7
- e = XlitEngine("gu", beam_width=10)
8
  embeddingfunc = embedding_functions.HuggingFaceEmbeddingFunction(api_key=os.environ["hf_token"],model_name="sentence-transformers/all-MiniLM-L6-v2")
9
  elibbookAI = chromadb.HttpClient("https://shethjenil-chromadb-server.hf.space/",port=443).get_or_create_collection("jainebooks")
10
  def qna(query:str,limit:int=1)->list:
11
- return [i["contentimg"] for i in elibbookAI.query(embeddingfunc(requests.get(f"https://translate.googleapis.com/translate_a/single?client=gtx&sl=gu&tl=en&dt=t&q={e.translit_sentence(query)['gu']}").json()[0][0][0]),n_results=limit)["metadatas"][0]]
12
  gr.Interface(qna,[gr.Textbox(),gr.Slider(1, 4, value=1, label="Count",step=1)],gr.Gallery()).launch()
 
1
  import chromadb
2
  import requests
3
  import chromadb.utils.embedding_functions as embedding_functions
 
4
  import gradio as gr
5
  import os
 
6
  embeddingfunc = embedding_functions.HuggingFaceEmbeddingFunction(api_key=os.environ["hf_token"],model_name="sentence-transformers/all-MiniLM-L6-v2")
7
  elibbookAI = chromadb.HttpClient("https://shethjenil-chromadb-server.hf.space/",port=443).get_or_create_collection("jainebooks")
8
  def qna(query:str,limit:int=1)->list:
9
+ return [i["contentimg"] for i in elibbookAI.query(embeddingfunc(requests.get(f"https://translate.googleapis.com/translate_a/single?client=gtx&sl=gu&tl=en&dt=t&q={query}").json()[0][0][0]),n_results=limit)["metadatas"][0]]
10
  gr.Interface(qna,[gr.Textbox(),gr.Slider(1, 4, value=1, label="Count",step=1)],gr.Gallery()).launch()