Update app.py

app.py

@@ -1,17 +1,15 @@
-from gradio_client import Client
+from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+import requests
+from PIL import Image
 
-client = Client("duchaba/Friendly_Text_Moderation")
-
-def find_tox_level(user_message):
-    result = client.predict(
-    		msg = user_message, # msg="Hello!!",
-    		safer=0.02,
-    		api_name="/fetch_toxicity_level"
-    )
-    
-tox_level = find_tox_level("What is your name?")
-
-print(tox_level)
+processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
+model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
 
+# load image from the IAM dataset
+url = "https://fki.tic.heia-fr.ch/static/img/a01-122-02.jpg"
+image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 
+pixel_values = processor(image, return_tensors="pt").pixel_values
+generated_ids = model.generate(pixel_values)
 
+generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]