Update app.py

app.py

@@ -3,12 +3,22 @@ import requests
 from PIL import Image
 from io import BytesIO
 import os
+from huggingface_hub import InferenceClient
+API_TOKEN = os.getenv("HF_API_TOKEN")  # Ensure you've set this environment variable
+API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
+
+enhancer = InferenceClient(api_key =API_TOKEN)
 
+for message in client.chat_completion(
+	model="meta-llama/Llama-3.2-1B-Instruct",
+	messages=[{"role": "user", "content": "What is the capital of France?"}],
+	max_tokens=500,
+	stream=True,
+):
+    print(message.choices[0].delta.content, end="")
 # Load API Token from environment variable
-API_TOKEN = os.getenv("HF_API_TOKEN")  # Ensure you've set this environment variable
 
 # Hugging Face Inference API URL
-API_URL = "https://api-inference.huggingface.co/models/enhanceaiteam/Flux-uncensored"
 
 # Function to call Hugging Face API and get the generated image
 def generate_image(prompt):