dalybuilds commited on
Commit
4fbde53
·
verified ·
1 Parent(s): fdaf554

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -26,11 +26,12 @@ def display_filtered_artworks(artist, genre, style):
26
  filtered_results = filter_artworks(artist, genre, style)
27
  return [(r["image"], f"Title: {r['title']}\nArtist: {r['artist']}\nStyle: {r['style']}\nGenre: {r['genre']}") for r in filtered_results]
28
 
29
- # Chatbot functionality for museum guide persona
30
- chatbot = pipeline("text-generation", model="gpt-4")
31
 
32
  def museum_guide_query(prompt):
33
- return chatbot(prompt, max_length=100, num_return_sequences=1)[0]["generated_text"]
 
34
 
35
  # Gradio interfaces
36
  artwork_interface = gr.Interface(
@@ -65,4 +66,4 @@ def launch_app():
65
  demo.launch()
66
 
67
  if __name__ == "__main__":
68
- launch_app()
 
26
  filtered_results = filter_artworks(artist, genre, style)
27
  return [(r["image"], f"Title: {r['title']}\nArtist: {r['artist']}\nStyle: {r['style']}\nGenre: {r['genre']}") for r in filtered_results]
28
 
29
+ # Chatbot functionality for museum guide persona using a publicly available Hugging Face model
30
+ chatbot = pipeline("text-generation", model="gpt2") # Replace with a valid Hugging Face model
31
 
32
  def museum_guide_query(prompt):
33
+ response = chatbot(prompt, max_length=100, num_return_sequences=1)
34
+ return response[0]["generated_text"]
35
 
36
  # Gradio interfaces
37
  artwork_interface = gr.Interface(
 
66
  demo.launch()
67
 
68
  if __name__ == "__main__":
69
+ launch_app()