Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
31 |
|
32 |
def museum_guide_query(prompt):
|
33 |
-
|
|
|
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()
|