Update app.py

app.py

@@ -1,29 +1,14 @@
-
+import gradio as gr
 from transformers import AutoProcessor, AutoModelForImageTextToText
 
-# Step 1: Load the model and processor from Hugging Face
 processor = AutoProcessor.from_pretrained("enzer1992/AI-Guru")
 model = AutoModelForImageTextToText.from_pretrained("enzer1992/AI-Guru")
 
-# Step 2: Function to interact with the model
-def chat_with_model(input_text):
-    # Process the input text
-    inputs = processor(input_text, return_tensors="pt")
-    
-    # Generate the model's response
+def generate_text(image):
+    inputs = processor(images=image, return_tensors="pt")
     outputs = model.generate(**inputs)
-    
-    # Decode and return the response
-    response = processor.decode(outputs[0], skip_special_tokens=True)
-    return response
+    return processor.decode(outputs[0], skip_special_tokens=True)
 
-# Step 3: Chat loop
-print("Chat with the model! Type 'exit' to end.")
-while True:
-    user_input = input("You: ")
-    if user_input.lower() == "exit":
-        break
-    
-    response = chat_with_model(user_input)
-    print(f"Model: {response}")
+iface = gr.Interface(fn=generate_text, inputs=gr.Image(), outputs="text")
+iface.launch()