Eminda commited on
Commit
7a09717
·
1 Parent(s): b0fbf4e

model change:

Browse files
Files changed (4) hide show
  1. .env +1 -0
  2. .gitignore +1 -0
  3. app.py +3 -1
  4. requirements.txt +2 -1
.env ADDED
@@ -0,0 +1 @@
 
 
1
+ HUGGINGFACEHUB_API_TOKEN=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
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ venv
app.py CHANGED
@@ -1,10 +1,12 @@
1
  import gradio as gr
 
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("text-generation", model="TheBloke/PMC_LLAMA-7B-10-Epoch-SuperHOT-8K-GPTQ")
5
 
6
  def greet(name):
7
  return pipe('who is the president of USA')
8
 
 
9
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
10
  iface.launch()
 
1
  import gradio as gr
2
+ # Use a pipeline as a high-level helper
3
  from transformers import pipeline
4
 
5
+ pipe = pipeline("text-generation", model="mosaicml/mpt-7b-instruct", trust_remote_code=True)
6
 
7
  def greet(name):
8
  return pipe('who is the president of USA')
9
 
10
+ # greet('df')
11
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
12
  iface.launch()
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  transformers
2
  gradio
3
- torch
 
 
1
  transformers
2
  gradio
3
+ torch
4
+ einops