Spaces:
Runtime error
Runtime error
added access token
Browse files- human_eval.py +7 -8
human_eval.py
CHANGED
|
@@ -11,6 +11,8 @@ import random
|
|
| 11 |
import pandas as pd
|
| 12 |
from collections import defaultdict
|
| 13 |
|
|
|
|
|
|
|
| 14 |
def encode_image_to_base64(image_path):
|
| 15 |
"""Encode an image or GIF file to base64."""
|
| 16 |
with open(image_path, "rb") as file:
|
|
@@ -275,16 +277,13 @@ def create_battle_arena(dataset_path, is_gif, is_causal):
|
|
| 275 |
return demo
|
| 276 |
|
| 277 |
if __name__ == "__main__":
|
| 278 |
-
|
| 279 |
-
# load the existing dataset that contains outputs of the LMs
|
| 280 |
-
human_eval_dataset = load_dataset("atlasia/LM-Moroccan-Darija-Bench", split='test').to_csv('human_eval_dataset.csv')
|
| 281 |
-
|
| 282 |
-
# precision
|
| 283 |
-
torch_dtype = torch.float16
|
| 284 |
|
| 285 |
-
# inference device
|
| 286 |
-
device = "cpu" #"cuda" if torch.cuda.is_available() else "cpu"
|
| 287 |
dataset_path = 'human_eval_dataset.csv'
|
| 288 |
is_gif = True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 289 |
demo = create_battle_arena(dataset_path, is_gif, is_causal=False)
|
| 290 |
demo.launch(debug=True)
|
|
|
|
| 11 |
import pandas as pd
|
| 12 |
from collections import defaultdict
|
| 13 |
|
| 14 |
+
TOKEN = os.environ['TOKEN']
|
| 15 |
+
|
| 16 |
def encode_image_to_base64(image_path):
|
| 17 |
"""Encode an image or GIF file to base64."""
|
| 18 |
with open(image_path, "rb") as file:
|
|
|
|
| 277 |
return demo
|
| 278 |
|
| 279 |
if __name__ == "__main__":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 280 |
|
|
|
|
|
|
|
| 281 |
dataset_path = 'human_eval_dataset.csv'
|
| 282 |
is_gif = True
|
| 283 |
+
|
| 284 |
+
# load the existing dataset that contains outputs of the LMs
|
| 285 |
+
human_eval_dataset = load_dataset("atlasia/LM-Moroccan-Darija-Bench", split='test', token=TOKEN).to_csv(dataset_path)
|
| 286 |
+
|
| 287 |
+
# load first tab for masked LM
|
| 288 |
demo = create_battle_arena(dataset_path, is_gif, is_causal=False)
|
| 289 |
demo.launch(debug=True)
|