deepapaikar
commited on
Commit
•
d7ed0bf
1
Parent(s):
ee6f18e
UPdating app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
|
|
7 |
torch.set_default_device("cuda")
|
8 |
|
9 |
|
10 |
-
model = "deepapaikar/
|
11 |
|
12 |
pipeline = transformers.pipeline(
|
13 |
"text-generation",
|
@@ -40,11 +40,11 @@ def gradio_predict(question, token):
|
|
40 |
# Define the Gradio interface
|
41 |
iface = gr.Interface(
|
42 |
fn=gradio_predict,
|
43 |
-
inputs=[gr.Textbox(label="Question", placeholder="e.g.
|
44 |
gr.Slider(2, 100, value=25, label="Token Count", info="Choose between 2 and 100")],
|
45 |
outputs=gr.TextArea(label="Answer"),
|
46 |
title="KatzBot",
|
47 |
-
description="
|
48 |
)
|
49 |
|
50 |
# Launch the app
|
|
|
7 |
torch.set_default_device("cuda")
|
8 |
|
9 |
|
10 |
+
model = "deepapaikar/katzbot-phi2"
|
11 |
|
12 |
pipeline = transformers.pipeline(
|
13 |
"text-generation",
|
|
|
40 |
# Define the Gradio interface
|
41 |
iface = gr.Interface(
|
42 |
fn=gradio_predict,
|
43 |
+
inputs=[gr.Textbox(label="Question", placeholder="e.g. Where is Yeshiva University located?", scale=4),
|
44 |
gr.Slider(2, 100, value=25, label="Token Count", info="Choose between 2 and 100")],
|
45 |
outputs=gr.TextArea(label="Answer"),
|
46 |
title="KatzBot",
|
47 |
+
description="Phi2-trial1",
|
48 |
)
|
49 |
|
50 |
# Launch the app
|