Update app.py

app.py

@@ -24,6 +24,11 @@ from spaces_info import description, examples, initial_prompt_value
 API_URL = os.getenv("API_URL")
 HF_API_TOKEN = os.getenv("HF_API_TOKEN")
 
+##Bloom Inference API
+API_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
+HF_TOKEN = os.environ["HF_TOKEN"]
+headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+
 print(API_URL)
 print(HF_API_TOKEN)
 
@@ -196,7 +201,8 @@ def classify_image(image):
 
 image = gr.inputs.Image(type="pil")
 label = gr.outputs.Label(num_top_classes=5)
-examples = [ ["cats.jpg"], ["batter.jpg"],["drinkers.jpg"] ] 
+#examples = [ ["cats.jpg"], ["batter.jpg"],["drinkers.jpg"] ]
+examples = [ ["batter.jpg"] ] 
 title = "Generate a Story from an Image"
 description = "Demo for classifying images with Perceiver IO. To use it, simply upload an image and click 'submit', a story is autogenerated as well"
 article = "<p style='text-align: center'></p>"