Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,10 +14,10 @@
|
|
14 |
# Use a pipeline as a high-level helper
|
15 |
from transformers import pipeline
|
16 |
|
17 |
-
|
18 |
|
19 |
def get_class(input):
|
20 |
-
return
|
21 |
|
22 |
iface = gr.Interface(fn=get_class, inputs="text", outputs=['text'], title='hello', description='LLM')
|
23 |
iface.launch()
|
|
|
14 |
# Use a pipeline as a high-level helper
|
15 |
from transformers import pipeline
|
16 |
|
17 |
+
my_pipe = pipeline("text-generation", model="NousResearch/Yarn-Mistral-7b-128k")
|
18 |
|
19 |
def get_class(input):
|
20 |
+
return my_pipe(input)
|
21 |
|
22 |
iface = gr.Interface(fn=get_class, inputs="text", outputs=['text'], title='hello', description='LLM')
|
23 |
iface.launch()
|