first push

README.md

@@ -1,13 +1,13 @@
 ---
 title: Indian Food Translator
-emoji: 📊
-colorFrom: red
-colorTo: indigo
+emoji: 
+colorFrom: orange
+colorTo: green
 sdk: gradio
-sdk_version: 3.0.26
+sdk_version: 3.0.22
 app_file: app.py
 pinned: false
-license: artistic-2.0
+license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,38 @@
+import os
+import gradio as gr
+from PIL import Image
+
+##Image Classification
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+extractor = AutoFeatureExtractor.from_pretrained("rajistics/finetuned-indian-food")
+model = AutoModelForImageClassification.from_pretrained("rajistics/finetuned-indian-food")
+
+def image_to_text(imagepic):
+  inputs = extractor(images=imagepic, return_tensors="pt")
+  outputs = model(**inputs)
+  logits = outputs.logits
+  predicted_class_idx = logits.argmax(-1).item()
+  return (model.config.id2label[predicted_class_idx])
+
+##Translation
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+#Get list of language codes: https://github.com/facebookresearch/flores/tree/main/flores200#languages-in-flores-200
+modelt = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
+tokenizert = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
+
+def translation(text):
+  translator = pipeline('translation', model=modelt, tokenizer=tokenizert, src_lang="eng_Latn", tgt_lang='ron_Latn')
+  output = translator(text)
+  return (output[0]['translation_text'])
+
+##Translation
+demo = gr.Blocks()
+with demo:
+    image_file = gr.inputs.Image(type="pil")
+    b1 = gr.Button("Recognize Image")
+    text = gr.Textbox()
+    b1.click(image_to_text, inputs=image_file, outputs=text)
+    b2 = gr.Button("Translation")
+    out1 = gr.Textbox()
+    b2.click(translation, inputs=text, outputs=out1)
+demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+#python-dotenv
+#protobuf~=3.19.0
+git+https://github.com/huggingface/transformers.git@33028f4c795e76f9e97226fc591bc7d0b8c7d815
+gradio
+Pillow