Spaces:
Runtime error
Runtime error
lhl
commited on
Commit
•
bc897bf
1
Parent(s):
9cacdb6
Update app.py
Browse files
app.py
CHANGED
@@ -3,11 +3,17 @@
|
|
3 |
from transformers import pipeline
|
4 |
import gradio as gr
|
5 |
|
6 |
-
|
7 |
|
|
|
8 |
demo = gr.Interface.from_pipeline(pipe)
|
9 |
demo.launch()
|
10 |
|
|
|
|
|
|
|
|
|
|
|
11 |
"""
|
12 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
13 |
import gradio as gr
|
|
|
3 |
from transformers import pipeline
|
4 |
import gradio as gr
|
5 |
|
6 |
+
model = "TinyLlama/TinyLlama-1.1B-Chat-v0.3"
|
7 |
|
8 |
+
pipe = pipeline("conversational", model=model)
|
9 |
demo = gr.Interface.from_pipeline(pipe)
|
10 |
demo.launch()
|
11 |
|
12 |
+
'''
|
13 |
+
demo = gr.Interface.load(model)
|
14 |
+
'''
|
15 |
+
|
16 |
+
|
17 |
"""
|
18 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
19 |
import gradio as gr
|