ArunAIML commited on
Commit
8681516
·
verified ·
1 Parent(s): 0b73e56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -1,15 +1,9 @@
1
- from langchain_nvidia_ai_endpoints import ChatNVIDIA
2
- from langchain_core.output_parsers import StrOutputParser
3
- from langchain_core.prompts import ChatPromptTemplate
4
  import gradio as gr
5
  import os
6
  from smolagents import HfApiModel
7
 
8
 
9
- prompt = ChatPromptTemplate.from_messages([("system", "You are a helpful AI assistant named Arun."), ("user", "{input}")])
10
-
11
- llm = ChatNVIDIA(model="mistralai/mixtral-8x7b-instruct-v0.1")
12
- chain = prompt | llm | StrOutputParser()
13
  model = HfApiModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1", token=os.environ.get("HF_TOKEN"))
14
 
15
  def chat(prompt, history):
 
1
+
 
 
2
  import gradio as gr
3
  import os
4
  from smolagents import HfApiModel
5
 
6
 
 
 
 
 
7
  model = HfApiModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1", token=os.environ.get("HF_TOKEN"))
8
 
9
  def chat(prompt, history):