Bils commited on
Commit
b5da3d3
·
verified ·
1 Parent(s): 3fcffdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoConfig, AutoModelForCausalLM
4
  from janus.models import MultiModalityCausalLM, VLChatProcessor
5
  from PIL import Image
6
  import numpy as np
7
- import spaces # Ensure this is available
8
 
9
  # Load the model and processor
10
  model_path = "deepseek-ai/Janus-Pro-7B"
@@ -99,7 +99,18 @@ def generate_image(prompt, seed=None, guidance=5, t2i_temperature=1.0):
99
  # Gradio interface
100
  def create_interface():
101
  with gr.Blocks() as demo:
102
- gr.Markdown("# Text-to-Image Generation")
 
 
 
 
 
 
 
 
 
 
 
103
 
104
  prompt_input = gr.Textbox(label="Prompt (describe the image)")
105
  seed_input = gr.Number(label="Seed (Optional)", value=12345, precision=0)
 
4
  from janus.models import MultiModalityCausalLM, VLChatProcessor
5
  from PIL import Image
6
  import numpy as np
7
+ import spaces # Import spaces for ZeroGPU compatibility
8
 
9
  # Load the model and processor
10
  model_path = "deepseek-ai/Janus-Pro-7B"
 
99
  # Gradio interface
100
  def create_interface():
101
  with gr.Blocks() as demo:
102
+ gr.Markdown("""
103
+ # Text-to-Image Generation with Janus-Pro-7B
104
+
105
+ Welcome to the Janus-Pro-7B Text-to-Image Generator! This advanced AI model by DeepSeek offers state-of-the-art capabilities in generating images from textual descriptions. Leveraging a unified multimodal framework, Janus-Pro-7B excels in both understanding and generating content, providing detailed and accurate visual representations based on your prompts.
106
+
107
+ **Key Features:**
108
+ - **Unified Multimodal Framework:** Integrates visual encoding and language processing for seamless understanding and generation.
109
+ - **High-Quality Image Generation:** Produces stable and detailed images that often surpass those from other leading models.
110
+ - **Flexible and Efficient:** Designed for a wide range of applications, from creative image generation to detailed visualizations.
111
+
112
+ For more information about Janus-Pro-7B, visit the [official Hugging Face model page](https://huggingface.co/deepseek-ai/Janus-Pro-7B).
113
+ """)
114
 
115
  prompt_input = gr.Textbox(label="Prompt (describe the image)")
116
  seed_input = gr.Number(label="Seed (Optional)", value=12345, precision=0)