tykiww commited on
Commit
cbd48ca
·
verified ·
1 Parent(s): b1cf10f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -15,11 +15,11 @@ def get_json_cfg():
15
 
16
  return config
17
 
18
- print(get_json_cfg())
19
 
20
 
21
- def greet(model_name, prompt_template, name):
22
- return f"Hello {name}!! Using model: {model_name} with template: {prompt_template}"
23
 
24
  model_choices = ["gpt2", "bert-base-uncased", "llama3-8b"]
25
 
@@ -35,7 +35,7 @@ with gr.Blocks() as demo:
35
  output = gr.Textbox(label="Output")
36
 
37
  greet_btn.click(fn=greet,
38
- inputs=[model_name, prompt_template, name_input],
39
  outputs=output)
40
 
41
 
 
15
 
16
  return config
17
 
18
+ #print(get_json_cfg())
19
 
20
 
21
+ def greet(model_name, prompt_template, name, dirname):
22
+ return f"Hello {name}!! Using model: {model_name} with template: {prompt_template} and {dirname}"
23
 
24
  model_choices = ["gpt2", "bert-base-uncased", "llama3-8b"]
25
 
 
35
  output = gr.Textbox(label="Output")
36
 
37
  greet_btn.click(fn=greet,
38
+ inputs=[model_name, prompt_template, name_input, os.getcwd()],
39
  outputs=output)
40
 
41