Spaces:
Running
Running
File size: 789 Bytes
9bfc384 fd0e912 c693731 9bfc384 c693731 9bfc384 fd0e912 9bfc384 7fb9cbe 9bfc384 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import streamlit as st
from transformers import pipeline
# Ensure the latest Transformers library is used
st.write("Initializing model...")
try:
pipe = pipeline("image-text-to-text", model="deepseek-ai/deepseek-vl2-small", trust_remote_code=True)
except ValueError as e:
st.error(f"Model loading failed: {e}")
st.stop()
st.title("DeepSeek-VL2 Chatbot")
# User input
user_input = st.text_input("Enter your message:", "")
if st.button("Generate Response"):
if user_input:
messages = [{"role": "user", "content": user_input}]
result = pipe(messages)
response = result[0]["generated_text"] if isinstance(result, list) else result
st.text_area("Model Response:", response, height=200)
else:
st.warning("Please enter a message.")
|