Update app.py

app.py

@@ -1 +1,38 @@
-import os from huggingface_hub import login, hf_hub_download from transformers import BlipProcessor, BlipForConditionalGeneration from PIL import Image import requests from io import BytesIO # توکن خود را از HuggingFace تنظیم کن os.environ["HF_HOME"] = "bermuda" # مسیر به توکن ایجاد شده در محیط # لاگین به Hugging Face Hub login(token="bermuda") # توکن خود را وارد کن # بارگیری مدل و پردازنده model_id = "Salesforce/blip2-flan-t5-small" # مدل BLIP برای پردازش تصویر # بارگذاری پردازشگر و مدل processor = BlipProcessor.from_pretrained(model_id) model = BlipForConditionalGeneration.from_pretrained(model_id) # بارگذاری و پردازش تصویر def image_to_text(image_url): # بارگیری تصویر از URL response = requests.get(image_url) image = Image.open(BytesIO(response.content)) # پردازش تصویر و تولید متن inputs = processor(images=image, return_tensors="pt") out = model.generate(**inputs) # تولید متن text = processor.decode(out[0], skip_special_tokens=True) return text # تست با یک تصویر image_url = "https://i1.delgarm.com/i/828/9910/20/5ff9472e5bb7a.jpg" # آدرس URL تصویر را وارد کن result = image_to_text(image_url) print("Generated Text:", result)
+import os
+from huggingface_hub import login, hf_hub_download
+from transformers import BlipProcessor, BlipForConditionalGeneration
+from PIL import Image
+import requests
+from io import BytesIO
+
+# توکن خود را از HuggingFace تنظیم کن
+os.environ["HF_HOME"] = "bermuda"  # مسیر به توکن ایجاد شده در محیط
+
+# لاگین به Hugging Face Hub
+login(token="bermuda")  # توکن خود را وارد کن
+
+# بارگیری مدل و پردازنده
+model_id = "Salesforce/blip2-flan-t5-small"  # مدل BLIP برای پردازش تصویر
+
+# بارگذاری پردازشگر و مدل
+processor = BlipProcessor.from_pretrained(model_id)
+model = BlipForConditionalGeneration.from_pretrained(model_id)
+
+# بارگذاری و پردازش تصویر
+def image_to_text(image_url):
+    # بارگیری تصویر از URL
+    response = requests.get(image_url)
+    image = Image.open(BytesIO(response.content))
+
+    # پردازش تصویر و تولید متن
+    inputs = processor(images=image, return_tensors="pt")
+    out = model.generate(**inputs)
+    
+    # تولید متن
+    text = processor.decode(out[0], skip_special_tokens=True)
+    return text
+
+# تست با یک تصویر
+image_url = "https://i1.delgarm.com/i/828/9910/20/5ff9472e5bb7a.jpg"  # آدرس URL تصویر را وارد کن
+result = image_to_text(image_url)
+print("Generated Text:", result)