RanM commited on
Commit
1819f08
·
verified ·
1 Parent(s): 52a3755

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  from io import BytesIO
3
  from PIL import Image
4
- from diffusers import AutoPipelineForText2Image
5
  import gradio as gr
6
  from generate_prompts import generate_prompt
7
  import base64
@@ -44,7 +44,6 @@ def generate_image(prompt):
44
 
45
  def inference(sentence_mapping, character_dict, selected_style):
46
  try:
47
- # Debugging statements
48
  print(f"Received sentence_mapping: {sentence_mapping}")
49
  print(f"Received character_dict: {character_dict}")
50
  print(f"Received selected_style: {selected_style}")
 
1
  import os
2
  from io import BytesIO
3
  from PIL import Image
4
+ from transformers import AutoPipelineForText2Image
5
  import gradio as gr
6
  from generate_prompts import generate_prompt
7
  import base64
 
44
 
45
  def inference(sentence_mapping, character_dict, selected_style):
46
  try:
 
47
  print(f"Received sentence_mapping: {sentence_mapping}")
48
  print(f"Received character_dict: {character_dict}")
49
  print(f"Received selected_style: {selected_style}")