import streamlit as st

from transformers import pipeline

try:
    
    messages = [
        {"role": "user", "content": "Who are you?"},
    ]
    pipe = pipeline("text-generation", model="deepseek-ai/deepseek-vl2-tiny", trust_remote_code=True)
    response = pipe(messages)
except:
    response = False
if response:
    st.write('Model imported successfully')
    print('Done')