GenAICoder commited on
Commit
76b397c
·
verified ·
1 Parent(s): aaf4aaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,10 +1,11 @@
1
  import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
- model_name = "deepseek-ai/DeepSeek-V3-Base"
5
  tokenizer = AutoTokenizer.from_pretrained(model_name)
6
  model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True,quantization_config=None)
7
 
 
8
  st.title("DeepSeek Chatbot")
9
 
10
  prompt = st.text_input("Enter your message:")
 
1
  import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
+ #model_name = "unsloth/DeepSeek-V3"
5
  tokenizer = AutoTokenizer.from_pretrained(model_name)
6
  model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True,quantization_config=None)
7
 
8
+
9
  st.title("DeepSeek Chatbot")
10
 
11
  prompt = st.text_input("Enter your message:")