mskov commited on
Commit
34ad65a
·
1 Parent(s): 0e57848

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -13,6 +13,7 @@ import torch
13
  from transformers import AutoModelForCausalLM
14
  from transformers import AutoTokenizer
15
  import time
 
16
 
17
 
18
  EXAMPLE_PROMPT = """This is a tool for helping someone with memory issues remember the next word.
@@ -63,7 +64,8 @@ def inference(audio, prompt, model, temperature, latest):
63
  max_tokens=8,
64
  n=5)
65
 
66
- infers = []
 
67
  temp = []
68
  #infered=[]
69
 
 
13
  from transformers import AutoModelForCausalLM
14
  from transformers import AutoTokenizer
15
  import time
16
+ import pandas as pd
17
 
18
 
19
  EXAMPLE_PROMPT = """This is a tool for helping someone with memory issues remember the next word.
 
64
  max_tokens=8,
65
  n=5)
66
 
67
+ #infers = []
68
+ infers = pd.DataFrame(columns=['One', 'Two', "Three", 'Four', 'Five'])
69
  temp = []
70
  #infered=[]
71