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
st.title('Code Generation')
huggingface_dataset_name = "red1xe/code_instructions"

dataset = load_dataset(huggingface_dataset_name)

st.write("## Dataset")
st.write(dataset)