update main

main.py

@@ -9,6 +9,7 @@ import io
 import base64
 import matplotlib.pyplot as plt 
 import torch
+import tensorflow
 from transformers import pipeline,VisionEncoderDecoderModel,ViTImageProcessor,AutoTokenizer
 from transformers import BartForConditionalGeneration, BartTokenizer
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -92,7 +93,17 @@ def index(req:Request):
 def index(req:Request):
     return  templates.TemplateResponse('ImageInterpretation.html',{'request':req})
 
+app.post("/caption2")
+async def generate_caption(file: UploadFile = File(...)):
+    contents = await file.read()
+    image = Image.open(io.BytesIO(contents)).convert("RGB")
 
+    # توليد caption
+    pixel_values = interpreter_processor(images=image, return_tensors="pt").pixel_values
+    output_ids = interpreter_model.generate(pixel_values, max_length=16, num_beams=4)
+    caption = interpreter_tokenizer.decode(output_ids[0], skip_special_tokens=True)
+
+    return {"caption": caption}
 
 @app.post("/caption")
 def caption(file:UploadFile=File(...)):
@@ -122,8 +133,6 @@ def summerzation(file:UploadFile=File(...)):
         text = get_text_from_EXCEL(file.file)
     else:
         return {"error": "Unsupported file type"}
-    if not text.strip():
-      return {"error": "File is empty"}
     
     result=""
     for i in range(0,len(text),1024):