ejschwartz commited on
Commit
cbfae43
·
1 Parent(s): 7f1d1fc
Files changed (3) hide show
  1. README.md +5 -1
  2. app.py +3 -1
  3. requirements.txt +1 -0
README.md CHANGED
@@ -9,4 +9,8 @@ app_file: app.py
9
  pinned: false
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
 
9
  pinned: false
10
  ---
11
 
12
+ # Aidapal Space
13
+
14
+ ## TODO
15
+
16
+ * `transformers` de-quantizes the gguf. We probably don't want that.
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  import spaces
3
  import torch
@@ -23,5 +24,6 @@ pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
23
  def greet(name):
24
  return pipe(name)
25
 
26
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
27
  demo.launch()
 
1
+ import frontmatter
2
  import gradio as gr
3
  import spaces
4
  import torch
 
24
  def greet(name):
25
  return pipe(name)
26
 
27
+ demo = gr.Interface(fn=greet, inputs="text", outputs="text",
28
+ description=frontmatter.load("README.md").content)
29
  demo.launch()
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  accelerate
2
  gguf
 
3
  torch
4
  transformers
 
1
  accelerate
2
  gguf
3
+ python-frontmatter
4
  torch
5
  transformers