Hugging Face's logo

Spaces:
yash-srivastava19
/
insta_captions
Runtime error

App Files Community
1
yash-srivastava19 commited on
Commit
d89154e
·
1 Parent(s): 6702c8a

Update app.py

Browse files

chamber of secrets

Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,8 +1,11 @@
 
1
  import cohere
2
  import gradio as gr
3
  from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
4
 
5
- co_client = cohere.Client('29JdDGuDUqPx2jqTkQUtsJqZRIwUoqwPKd2j9CRA')
 
 
6
 
7
  device = 'cpu'
8
  encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
 
1
+ import os
2
  import cohere
3
  import gradio as gr
4
  from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel
5
 
6
+ COHERE_API_KEY = os.getenv['COHERE_API_KEY']
7
+
8
+ co_client = cohere.Client(COHERE_API_KEY)
9
 
10
  device = 'cpu'
11
  encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"