Raybilhelp's picture
Update app.py
30d1f93 verified
Raw
History Blame Contribute Delete
1.61 kB
import os
import gradio as gr
import torch
from diffusers import StableDiffusionPipeline
# মডেল আইডি এবং পাইপলাইন সেটআপ (১৬ জিবি র‍্যাম ও CPU-এর জন্য অপ্টিমাইজড)
model_id = "OFA-Sys/small-stable-diffusion-v0"
# torch_dtype এর জায়গায় সঠিক নিয়ম (dtype) ব্যবহার করা হলো অবসোলেট এরর এড়াতে
pipe = StableDiffusionPipeline.from_pretrained(model_id, dtype=torch.float32)
pipe = pipe.to("cpu")
def generate_video_frame(prompt):
# প্রম্পট দিয়ে ইমেজ ফ্রেম তৈরি
image = pipe(prompt, num_inference_steps=15).images[0]
return image
# গ্রাডিও ইউজার ইন্টারফেস (মোবাইল ফ্রেন্ডলি)
interface = gr.Interface(
fn=generate_video_frame,
inputs=gr.Textbox(label="আপনার এআই প্রম্পট লিখুন", placeholder="A running cat, cinematic..."),
outputs=gr.Image(label="Generated Output"),
title="Rayhan's Free AI Space",
description="এই স্পেসটি সম্পূর্ণ ফ্রিতে এবং আনলিমিটেড বার ব্যবহার করতে পারবেন।"
)
if __name__ == "__main__":
# ভুল 'docker_setting' আর্গুমেন্টটি ফেলে দিয়ে ক্লিন কোড করা হলো
interface.launch(server_name="0.0.0.0", server_port=7860)