File size: 400 Bytes
da172aa
 
 
 
 
 
 
2856ccc
 
da172aa
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from datasets import load_dataset
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, GenerationConfig, TrainingArguments, Trainer
import torch
import time
import evaluate
import pandas as pd
import numpy as np
import streamlit as st

huggingface_dataset_name = "red1xe/code_instructions"

dataset = load_dataset(huggingface_dataset_name)

st.spinner("Loading model...")
st.write(dataset)