Fangyu Liu commited on
Commit
f9e72d0
1 Parent(s): 8a49fd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -2,11 +2,11 @@ import gradio as gr
2
  import requests
3
  from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
4
  import os
 
 
 
5
 
6
- ##Bloom
7
- API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
8
- HF_API_TOKEN = os.getenv("HF_API_TOKEN")
9
- headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
10
 
11
  def _add_markup(table):
12
  parts = [p.strip() for p in table.splitlines(keepends=False)]
@@ -49,10 +49,7 @@ A: Let's find the row of year 2007, that's Row 3. Let's extract the numbers on R
49
  {_INSTRUCTION}"""
50
 
51
 
52
- import torch
53
- from peft import PeftModel
54
- import transformers
55
- import gradio as gr
56
 
57
  assert (
58
  "LlamaTokenizer" in transformers._import_structure["models.llama"]
 
2
  import requests
3
  from transformers import Pix2StructForConditionalGeneration, Pix2StructProcessor
4
  import os
5
+ import torch
6
+ from peft import PeftModel
7
+ import transformers
8
 
9
+ ## CoT prompts
 
 
 
10
 
11
  def _add_markup(table):
12
  parts = [p.strip() for p in table.splitlines(keepends=False)]
 
49
  {_INSTRUCTION}"""
50
 
51
 
52
+ ## alpaca-lora
 
 
 
53
 
54
  assert (
55
  "LlamaTokenizer" in transformers._import_structure["models.llama"]