richardkimsm89 commited on
Commit
c3d5504
·
verified ·
1 Parent(s): 0779167

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,10 +1,11 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
- pipeline = pipeline(model = "meta-llama/Llama-3.2-1B-Instruct")
 
5
 
6
  def inference(input):
7
- output = pipeline(input)
8
  return output[0]['generated_text']
9
 
10
  app = gr.Interface(
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ pipe1 = pipeline(model = "meta-llama/Llama-3.2-1B-Instruct")
5
+ #pipe2 = pipeline(model = "meta-llama/Llama-3.2-3B-Instruct")
6
 
7
  def inference(input):
8
+ output = pipe1(input)
9
  return output[0]['generated_text']
10
 
11
  app = gr.Interface(