Spaces:
Paused
Paused
import streamlit as st | |
from model_utils import load_model, create_pipeline, generate_text | |
# Load model and tokenizer | |
st.write("Loading model...") | |
model, tokenizer = load_model() | |
hf_pipeline = create_pipeline(model, tokenizer) | |
st.write("Model loaded successfully!") | |
# Streamlit application | |
st.title("Hugging Face Model with LangChain and Streamlit") | |
# Text input field | |
input_text = st.text_input("Enter text to translate:") | |
# Button to generate response | |
if st.button("Translate"): | |
if input_text: | |
# Generate response using LangChain | |
response = generate_text(hf_pipeline, input_text) | |
# Display the response | |
st.write("Translated text:", response) | |
else: | |
st.write("Please enter some text.") | |