pratikshahp commited on
Commit
d8b8743
·
verified ·
1 Parent(s): 872a142

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,7 +2,6 @@ import os
2
  from dotenv import load_dotenv
3
  import gradio as gr
4
  from langchain_huggingface import HuggingFaceEndpoint
5
- from langchain_core.messages import HumanMessage
6
 
7
  # Load environment variables
8
  load_dotenv()
@@ -10,7 +9,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
10
 
11
  # Initialize the HuggingFace inference endpoint
12
  llm = HuggingFaceEndpoint(
13
- repo_id="flax-community/t5-recipe-generation",
14
  huggingfacehub_api_token=HF_TOKEN.strip(),
15
  temperature=0.7,
16
  max_new_tokens=150
 
2
  from dotenv import load_dotenv
3
  import gradio as gr
4
  from langchain_huggingface import HuggingFaceEndpoint
 
5
 
6
  # Load environment variables
7
  load_dotenv()
 
9
 
10
  # Initialize the HuggingFace inference endpoint
11
  llm = HuggingFaceEndpoint(
12
+ repo_id="mistralai/Mistral-7B-Instruct-v0.3,
13
  huggingfacehub_api_token=HF_TOKEN.strip(),
14
  temperature=0.7,
15
  max_new_tokens=150