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)