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