Spaces:
Sleeping
Sleeping
| import os | |
| import streamlit as st | |
| from huggingface_hub import InferenceClient | |
| api_key = os.getenv("HF_API_KEY") | |
| client = InferenceClient(api_key=api_key) | |
| user_input = st.text_input('Ask me a question') | |
| messages = [{"role": "user","content": user_input}] | |
| completion = client.chat.completions.create( | |
| model="Qwen/Qwen2.5-Coder-32B-Instruct", | |
| messages=messages, | |
| max_tokens=500 | |
| ) | |
| if st.button("Submit"): | |
| if user_input: | |
| response = completion.choices[0].message | |
| st.write(f"Adrega AI: {response}") | |
| else: | |
| st.write("Please enter a question.") | |
| print(completion.choices[0].message) |