Upload app.py

app.py

@@ -42,6 +42,11 @@ phi2 = pipeline(
 )
 
 def generate(message, max_new_tokens):
+
+  # Provide a default value if max_new_tokens is None
+    if max_new_tokens is None:
+        max_new_tokens = 50  # Default value; adjust as needed based on your typical use case
+
     instruction = "You are a helpful assistant to 'User'. You do not respond as 'User' or pretend to be 'User'. You only respond once as 'Assistant'."
     final_prompt = f"Instruction: {instruction}\nUser: {message}\nOutput:"