tykiww commited on
Commit
5d52f32
1 Parent(s): 1e282db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,9 +3,12 @@ import gradio as gr
3
  def greet(name):
4
  return "Hello " + name + "!!"
5
 
 
 
 
6
  with gr.Blocks() as demo:
7
  gr.Markdown("# Instruction Tuning with Unsloth")
8
-
9
  gr.Interface(fn=greet,
10
  inputs="text",
11
  outputs="text")
 
3
  def greet(name):
4
  return "Hello " + name + "!!"
5
 
6
+ model_choices = ["gpt2", "bert-base-uncased", "llama3-8b"]
7
+
8
+
9
  with gr.Blocks() as demo:
10
  gr.Markdown("# Instruction Tuning with Unsloth")
11
+ model_name = gr.Dropdown(label="Model", choices=model_choices, value="gpt2")
12
  gr.Interface(fn=greet,
13
  inputs="text",
14
  outputs="text")