kartik91 commited on
Commit
ea4c8eb
·
verified ·
1 Parent(s): aca5479

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,11 +8,11 @@ pipe2 = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
8
  repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
9
 
10
 
11
- llm = HuggingFacePipeline.from_model_id(
12
- model_id=repo_id,
13
- task="text-generation",
14
- pipeline_kwargs={"max_length": 100,"temperature":0.5},
15
- )
16
  def reduce_add(a):
17
  ll=dict()
18
  for i in a:
 
8
  repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
9
 
10
 
11
+ llm = HuggingFaceEndpoint(
12
+ repo_id=repo_id,
13
+ max_length=100,
14
+ temperature=0.5
15
+ )
16
  def reduce_add(a):
17
  ll=dict()
18
  for i in a: