import streamlit as st
from transformers import AutoTokenizer, AutoModelForCausalLM

# Load your model here
tokenizer = AutoTokenizer.from_pretrained("qbwmwsap/amber-model-mine")
model = AutoModelForCausalLM.from_pretrained("qbwmwsap/amber-model-mine")

st.title('Amber-Model-Remastered')

text = st.text_input('Enter some text')

if text:
    # Preprocess your data here
    input_ids = tokenizer(text, return_tensors="pt").input_ids

    # Run the model and get the output
    outputs = model.generate(input_ids)

    # Postprocess your output here if necessary    
    output = tokenizer.decode(outputs[0])

    st.write(output)