t2ag3 commited on
Commit
1099265
·
verified ·
1 Parent(s): 3eb3b95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -12,7 +12,6 @@ For more information on `huggingface_hub` Inference API support, please check th
12
  """
13
 
14
  model_name = "llama-3.3-70b-versatile"
15
- groq_api_key = "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"
16
  groq_chat = ChatGroq(groq_api_key=groq_api_key, model_name=model_name)
17
  embeddings = HuggingFaceEmbeddings(
18
  model_name = "pkshatech/GLuCoSE-base-ja"
@@ -23,7 +22,7 @@ vector_store = InMemoryVectorStore.load(
23
  retriever = vector_store.as_retriever(search_kwargs={"k": 4})
24
 
25
 
26
- def fetch_response(user_input):
27
  chat = ChatGroq(
28
  api_key = groq_api_key,
29
  model_name = model_name
@@ -57,6 +56,8 @@ with gr.Blocks() as demo:
57
  gr.Markdown('''# SOP事業マスター \n
58
  SOP作成研究に関して、公募要領やQAを参考にRAGを使って回答します。
59
  ''')
 
 
60
  with gr.Row():
61
  with gr.Column():
62
  user_input = gr.Textbox(label="User Input")
@@ -67,7 +68,7 @@ with gr.Blocks() as demo:
67
  source1 = gr.Textbox(label="回答ソース1")
68
  with gr.Column():
69
  source2 = gr.Textbox(label="回答ソース2")
70
- submit.click(fetch_response, inputs=user_input, outputs=[answer, source1, source2])
71
 
72
  if __name__ == "__main__":
73
  demo.launch()
 
12
  """
13
 
14
  model_name = "llama-3.3-70b-versatile"
 
15
  groq_chat = ChatGroq(groq_api_key=groq_api_key, model_name=model_name)
16
  embeddings = HuggingFaceEmbeddings(
17
  model_name = "pkshatech/GLuCoSE-base-ja"
 
22
  retriever = vector_store.as_retriever(search_kwargs={"k": 4})
23
 
24
 
25
+ def fetch_response(groq_api_key, user_input):
26
  chat = ChatGroq(
27
  api_key = groq_api_key,
28
  model_name = model_name
 
56
  gr.Markdown('''# SOP事業マスター \n
57
  SOP作成研究に関して、公募要領やQAを参考にRAGを使って回答します。
58
  ''')
59
+ with gr.Row():
60
+ api_key = gr.Textbox(label="Groq API key")
61
  with gr.Row():
62
  with gr.Column():
63
  user_input = gr.Textbox(label="User Input")
 
68
  source1 = gr.Textbox(label="回答ソース1")
69
  with gr.Column():
70
  source2 = gr.Textbox(label="回答ソース2")
71
+ submit.click(fetch_response, inputs=[api_key, user_input], outputs=[answer, source1, source2])
72
 
73
  if __name__ == "__main__":
74
  demo.launch()