Update app.py
Browse files
app.py
CHANGED
@@ -103,8 +103,11 @@ def generate_response(prompt):
|
|
103 |
|
104 |
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
105 |
|
106 |
-
return (
|
107 |
-
f"
|
|
|
|
|
|
|
108 |
|
109 |
# Define chat function for gr.ChatInterface
|
110 |
def chat_function(message, history):
|
|
|
103 |
|
104 |
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
105 |
|
106 |
+
return (
|
107 |
+
f"Movie Info: {movie_info['title']}, {movie_info['year']}, {movie_info['genre']}\n\n" \
|
108 |
+
f"Generated Response: {generated_text[:100]}", # Truncate to 100 characters
|
109 |
+
image_url
|
110 |
+
)
|
111 |
|
112 |
# Define chat function for gr.ChatInterface
|
113 |
def chat_function(message, history):
|