ashwinR commited on
Commit
d00d793
1 Parent(s): 27c9822

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import (
7
  pipeline,
8
  )
9
 
10
- model_name = "sagard21/python-code-explainer"
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_name, padding=True)
13
 
@@ -44,6 +44,6 @@ if __name__ == "__main__":
44
  with gr.Blocks() as demo:
45
  gr.Interface(fn = generate_text, inputs = textbox1, outputs = textbox2)
46
  with gr.Row():
47
- gr.Image(value = "output.jpg", label = "Sample Explaination in Natural Language")
48
- gr.Image(value = "code.jpg", label = "Sample Code for Checking if a Binary Tree is Mirrored")
49
  demo.launch()
 
7
  pipeline,
8
  )
9
 
10
+ model_name = "ashwinR/CodeExplainer"
11
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_name, padding=True)
13
 
 
44
  with gr.Blocks() as demo:
45
  gr.Interface(fn = generate_text, inputs = textbox1, outputs = textbox2)
46
  with gr.Row():
47
+ gr.Image(value = "output.jpg", label = "Sample Code for Checking if a Binary Tree is Mirrored")
48
+ gr.Image(value = "code.jpg", label = "Sample Output Explaination in Natural language")
49
  demo.launch()