Update app.py

app.py

@@ -63,7 +63,12 @@ def generate_tasks_from_model(api_url, num_tasks):
             outputs = model.generate(**inputs, max_length=50)  # Limit max length for speed
             task = tokenizer.decode(outputs[0], skip_special_tokens=True)
             print(f"Generated task: {task}")
-            tasks.append(task)
+            # Check if the generated task is a valid URL
+            if task.startswith("http://") or task.startswith("https://"):
+                tasks.append(task)
+            else:
+                print(f"Invalid task generated: {task}")
+                tasks.append(f"Error: Invalid task generated: {task}")
         except Exception as e:
             print(f"Error generating task: {str(e)}")
             tasks.append(f"Error generating task: {str(e)}")
@@ -125,4 +130,4 @@ iface = gr.Interface(
     """
 )
 
-iface.launch()
+iface.launch()