codeGPT / app.py
red1xe's picture
Add application file
7f51a5a
raw
history blame
420 Bytes
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)