Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 from transformers import MllamaForConditionalGeneration, AutoProcessor
 from huggingface_hub import login
 import io
+
 # Authenticate with Hugging Face
 HF_TOKEN = st.secrets["newfinegrained"]
 login(HF_TOKEN)
@@ -22,9 +23,16 @@ def load_model_and_processor(model_id):
 def generate_text(model, processor, image_url, prompt):
     """Generate text using the model and processor."""
     try:
+        # Fetch the image from the URL
+        response = requests.get(image_url)
+        response.raise_for_status()  # Raise an error if the request fails
         image = Image.open(io.BytesIO(response.content))
+
+        # Process the image and prompt
         inputs = processor(image, prompt, return_tensors="pt").to(model.device)
         output = model.generate(**inputs, max_new_tokens=30)
+
+        # Decode the output
         return processor.decode(output[0])
     except Exception as e:
         return f"Error: {e}"
@@ -64,6 +72,7 @@ if st.button("Generate Haiku"):
 
 
 
+