screenshot to code ui

app.py

@@ -1,9 +1,25 @@
+from tkinter import Image
 import streamlit as st
+from transformers import pipeline
 
-# Write a title and some text to the app:
-st.title('My Hugging Face Space')
-st.write('This is a simple Streamlit app in a Hugging Face Space!')
+model_name = "HuggingFaceM4/VLM_WebSight_finetuned"
+image_to_code = pipeline("image-to-text", model=model_name)
+
+st.title("Image to Code Converter")
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png"])
+
+if uploaded_file is not None:
+    # To see details
+    st.image(uploaded_file, caption='Uploaded Image.', use_column_width=True)
+    st.write("")
+    st.write("Converting image to code...")
+
+    # Convert the file to an image
+    image = Image.open(uploaded_file)
+
+    # Process the image
+    code_result = image_to_code(image)
+
+    # Display the code
+    st.code(code_result[0]['generated_text'])
 
-# Interactive widgets:
-number = st.slider('Select a number', 0, 10, 5)
-st.write('You selected:', number)

requirements.txt

@@ -1,3 +1,4 @@
 streamlit
 transformers
-torch
+torch
+tkinter