Miladata commited on
Commit
80dc2cf
·
verified ·
1 Parent(s): 4ea2a03

Feat: args and return methods added on the generation tool

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -71,6 +71,12 @@ def analyze_species_difference(species1: str, species2: str) -> str:
71
  def generate_starship_image(starship_class: str) -> str:
72
  """
73
  Generate an image of a Star Trek starship based on the starship class.
 
 
 
 
 
 
74
  """
75
  prompt = f"Generate a digital image of Star Trek starship of the {starship_class} class."
76
 
 
71
  def generate_starship_image(starship_class: str) -> str:
72
  """
73
  Generate an image of a Star Trek starship based on the starship class.
74
+
75
+ Args:
76
+ starship_class: The class of the starship to generate (e.g.,"Defiant-class", "Galaxy-class").
77
+
78
+ Returns:
79
+ A URL to the generated image of the specified starship class.
80
  """
81
  prompt = f"Generate a digital image of Star Trek starship of the {starship_class} class."
82