Anshini commited on
Commit
973212e
·
verified ·
1 Parent(s): 8e6e937

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import os
3
+ from langchain_huggingface import HuggingFaceEndpoint, HuggingFacePipeline, ChatHuggingFace
4
+ from langchain_core.messages import HumanMessage,SystemMessage,AIMessage
5
+
6
+ hf_token = os.getenv("HF_TOKEN")
7
+
8
+ deep_seek_model = HuggingFaceEndpoint(repo_id = 'deepseek-ai/DeepSeek-R1', provider = 'nebius', temperature = 0.7, max_new_tokens = 100, task ='conversational')
9
+ ds_model = ChatHuggingFace(llm = deep_seek_model, repo_id = 'deepseek-ai/DeepSeek-R1', provider = 'nebius', temperature = 0.7, max_new_tokens = 100, task ='conversational')
10
+
11
+ st.title("ChatGenius Hub: Master Every Data Skills")
12
+
13
+ message = [SystemMessage(content = "Answer like a data scientist"),
14
+ HumanMessage(content = "Tell me about Support Vector")]
15
+ result = ds_model.invole(message)
16
+ print(result.content)
17
+