Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|