GenAiPA commited on
Commit
60293f7
·
verified ·
1 Parent(s): f2818a2

Update app.py

Browse files

udpated the code after creting new secret key

Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
 
3
 
4
-
 
 
5
 
6
  def generate_response(prompt):
7
  response = client.text_generation(prompt)
@@ -16,10 +19,10 @@ with gr.Blocks() as demo:
16
  # Output component
17
  output = gr.Textbox(label="AI response")
18
 
19
- # Button to trigger generation
20
  submit_btn = gr.Button("Submit")
21
- # Button to trigger generation
22
- submit_btn.click(generate_response, inputs=msg, outputs=output)
23
 
24
- demo.launch()
 
25
 
 
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
+ import os
4
 
5
+ # Retrieve the token from an environment variable
6
+ token = os.getenv("PA_HF_TOKEN")
7
+ client = InferenceClient(model="gpt2", token=token)
8
 
9
  def generate_response(prompt):
10
  response = client.text_generation(prompt)
 
19
  # Output component
20
  output = gr.Textbox(label="AI response")
21
 
22
+ # Button to trigger generation
23
  submit_btn = gr.Button("Submit")
 
 
24
 
25
+ # Link the button to the function
26
+ submit_btn.click(generate_response, inputs=msg, outputs=output)
27
 
28
+ demo.launch()