eliwill's picture
Testing out loading in model
685f9f7
raw
history blame
541 Bytes
import gradio as gr
from transformers import pipeline
import numpy as np
import pandas as pd
from sentence_transformers import SentenceTransformer, util
import nltk
from nltk import sent_tokenize
nltk.download("punkt")
krishnamurti_generator = pipeline("text-generation", model="distilgpt2")
# stoic_generator = pipeline("text-generation", model="eliwill/stoic-generator-10e")
############### BUILDING DEMO ################################
with gr.Blocks() as demo:
gr.Markdown("""
# Ask a Philsopher
"""
)
demo.launch()