baconnier commited on
Commit
b68c9a6
·
verified ·
1 Parent(s): a217992

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -114,4 +114,15 @@ class GradioInterface:
114
  return f"Error: {str(e)}", f"Error: {str(e)}"
115
 
116
  def launch(self, share=False):
117
- self.interface.launch(share=share)
 
 
 
 
 
 
 
 
 
 
 
 
114
  return f"Error: {str(e)}", f"Error: {str(e)}"
115
 
116
  def launch(self, share=False):
117
+ self.interface.launch(share=share)
118
+
119
+
120
+ if __name__ == '__main__':
121
+ # Initialize the prompt refiner with API token
122
+ prompt_refiner = PromptRefiner(api_token)
123
+
124
+ # Create the Gradio interface
125
+ gradio_interface = GradioInterface(prompt_refiner, custom_css)
126
+
127
+ # Launch the interface
128
+ gradio_interface.launch(share=True)