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 |
|