lhl commited on
Commit
bc897bf
1 Parent(s): 9cacdb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -3,11 +3,17 @@
3
  from transformers import pipeline
4
  import gradio as gr
5
 
6
- pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
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