Spaces:
Runtime error
Runtime error
Chenyu Yang
commited on
Commit
•
18d5051
1
Parent(s):
da69d05
llama2
Browse files
app.py
CHANGED
@@ -1,15 +1,13 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
pipeline = pipeline(task="
|
5 |
|
6 |
-
def predict(
|
7 |
-
|
8 |
-
return {p["label"]: p["score"] for p in predictions}
|
9 |
|
10 |
gr.Interface(
|
11 |
-
predict,
|
12 |
-
inputs=gr.
|
13 |
-
outputs=
|
14 |
-
title="Hot Dog? Or Not?",
|
15 |
).launch()
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
|
4 |
+
pipeline = pipeline(task="text-generation", model="meta-llama/Llama-2-7b")
|
5 |
|
6 |
+
def predict(input):
|
7 |
+
return pipeline("I can't believe you did such a ", do_sample=False)
|
|
|
8 |
|
9 |
gr.Interface(
|
10 |
+
fn=predict,
|
11 |
+
inputs=gr.Textbox(lines=2, placeholder="Text here..."),
|
12 |
+
outputs="text",
|
|
|
13 |
).launch()
|